diff --git a/examples/multi_node_examples/job_submit.sh b/examples/multi_node_examples/job_submit.sh
index 62ddbca5..159d0df2 100755
--- a/examples/multi_node_examples/job_submit.sh
+++ b/examples/multi_node_examples/job_submit.sh
@@ -8,7 +8,7 @@
 #SBATCH --time=0-02:00:00
 
 # activate conda env
-source activate pl3
+source activate $1
 
 # -------------------------
 # debugging flags (optional)