diff --git a/examples/multi_node_examples/README.md b/examples/multi_node_examples/README.md
index d1d194f0..0db9d462 100644
--- a/examples/multi_node_examples/README.md
+++ b/examples/multi_node_examples/README.md
@@ -6,5 +6,5 @@ To run this demo which launches a single job that trains on 2 nodes (2 gpus per
 2. Create a conda environment with Lightning and a GPU PyTorch version.   
 3. Submit this script.   
 ```bash
-sbatch job_submit.sh your_env_name_with_lightning_installed
-```
\ No newline at end of file
+sbatch job_submit.sh --env=YourEnv
+```
diff --git a/examples/multi_node_examples/job_submit.sh b/examples/multi_node_examples/job_submit.sh
index ee737198..176bf09f 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
-conda activate $1
+source activate $env
 
 # -------------------------
 # debugging flags (optional)
@@ -24,4 +24,4 @@ conda activate $1
 # -------------------------
 
 # run script from above
-python multi_node_demo.py
\ No newline at end of file
+srun python multi_node_demo.py