diff --git a/tests/integration/test_code/test_code_beluga__salloc_.txt b/tests/integration/test_code/test_code_beluga__salloc_.txt
index ab393fc8..f47b278e 100644
--- a/tests/integration/test_code/test_code_beluga__salloc_.txt
+++ b/tests/integration/test_code/test_code_beluga__salloc_.txt
@@ -1,11 +1,12 @@
 Checking disk quota on $HOME...
 Disk usage: X / LIMIT GiB and X / LIMIT files
-Installing VSCode extensions that are on the local machine on beluga in the background.
+Installing VSCode extensions that are on the local machine on beluga.
 (beluga) $ cd $SCRATCH && salloc --wckey=milatools_test --account=SLURM_ACCOUNT --nodes=1 --ntasks=1 --cpus-per-task=1 --mem=1G --time=0:05:00 --oversubscribe --job-name=mila-code
 Waiting for job JOB_ID to start.
-(local) echo --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
+✓ Syncing vscode extensions:  N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS
+✓ localhost -> beluga - Done. N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS
+(local) $ echo --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
 --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
 
 The editor was closed. Reopen it with <Enter> or terminate the process with <Ctrl+C> (maybe twice).
-(beluga) $ scancel JOB_ID
 Ended session on 'COMPUTE_NODE'
\ No newline at end of file
diff --git a/tests/integration/test_code/test_code_beluga__sbatch_.txt b/tests/integration/test_code/test_code_beluga__sbatch_.txt
index 13cb7a53..289f9938 100644
--- a/tests/integration/test_code/test_code_beluga__sbatch_.txt
+++ b/tests/integration/test_code/test_code_beluga__sbatch_.txt
@@ -1,11 +1,13 @@
 Checking disk quota on $HOME...
 Disk usage: X / LIMIT GiB and X / LIMIT files
-Installing VSCode extensions that are on the local machine on beluga in the background.
-(beluga) $ cd $SCRATCH && sbatch --parsable --wckey=milatools_test --account=SLURM_ACCOUNT --nodes=1 --ntasks=1 --cpus-per-task=1 --mem=1G --time=0:05:00 --oversubscribe --job-name=mila-code --wrap
-'srun sleep 7d'
+Installing VSCode extensions that are on the local machine on beluga.
+(beluga) $ cd $SCRATCH && sbatch --parsable --wckey=milatools_test --account=SLURM_ACCOUNT --nodes=1 --ntasks=1 --cpus-per-task=1 --mem=1G --time=0:05:00 --oversubscribe
+--job-name=mila-code --wrap 'srun sleep 7d'
 JOB_ID
 
-(local) echo --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
+✓ Syncing vscode extensions:  N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS
+✓ localhost -> beluga - Done. N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS
+(local) $ echo --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
 --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
 
 The editor was closed. Reopen it with <Enter> or terminate the process with <Ctrl+C> (maybe twice).
diff --git a/tests/integration/test_code/test_code_graham__salloc_.txt b/tests/integration/test_code/test_code_graham__salloc_.txt
index 6c4d1df8..812dd3e6 100644
--- a/tests/integration/test_code/test_code_graham__salloc_.txt
+++ b/tests/integration/test_code/test_code_graham__salloc_.txt
@@ -1,10 +1,12 @@
 Checking disk quota on $HOME...
-Installing VSCode extensions that are on the local machine on graham in the background.
+Disk usage: X / LIMIT GiB and X / LIMIT files
+Installing VSCode extensions that are on the local machine on graham.
 (graham) $ cd $SCRATCH && salloc --wckey=milatools_test --account=SLURM_ACCOUNT --nodes=1 --ntasks=1 --cpus-per-task=1 --mem=1G --time=0:05:00 --oversubscribe --job-name=mila-code
 Waiting for job JOB_ID to start.
-(local) echo --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
+✓ Syncing vscode extensions:  N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS
+✓ localhost -> graham - Done. N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS
+(local) $ echo --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
 --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
 
 The editor was closed. Reopen it with <Enter> or terminate the process with <Ctrl+C> (maybe twice).
-(graham) $ scancel JOB_ID
 Ended session on 'COMPUTE_NODE'
\ No newline at end of file
diff --git a/tests/integration/test_code/test_code_graham__sbatch_.txt b/tests/integration/test_code/test_code_graham__sbatch_.txt
index dbcb1341..a90e7bfc 100644
--- a/tests/integration/test_code/test_code_graham__sbatch_.txt
+++ b/tests/integration/test_code/test_code_graham__sbatch_.txt
@@ -1,10 +1,13 @@
 Checking disk quota on $HOME...
-Installing VSCode extensions that are on the local machine on graham in the background.
-(graham) $ cd $SCRATCH && sbatch --parsable --wckey=milatools_test --account=SLURM_ACCOUNT --nodes=1 --ntasks=1 --cpus-per-task=1 --mem=1G --time=0:05:00 --oversubscribe --job-name=mila-code --wrap
-'srun sleep 7d'
+Disk usage: X / LIMIT GiB and X / LIMIT files
+Installing VSCode extensions that are on the local machine on graham.
+(graham) $ cd $SCRATCH && sbatch --parsable --wckey=milatools_test --account=SLURM_ACCOUNT --nodes=1 --ntasks=1 --cpus-per-task=1 --mem=1G --time=0:05:00 --oversubscribe
+--job-name=mila-code --wrap 'srun sleep 7d'
 JOB_ID
 
-(local) echo --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
+✓ Syncing vscode extensions:  N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS
+✓ localhost -> graham - Done. N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS
+(local) $ echo --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
 --new-window --wait --remote ssh-remote+COMPUTE_NODE $HOME/bob
 
 The editor was closed. Reopen it with <Enter> or terminate the process with <Ctrl+C> (maybe twice).
diff --git a/tests/integration/test_code/test_code_narval__sbatch_.txt b/tests/integration/test_code/test_code_narval__sbatch_.txt
index f4786455..12d19568 100644
--- a/tests/integration/test_code/test_code_narval__sbatch_.txt
+++ b/tests/integration/test_code/test_code_narval__sbatch_.txt
@@ -1,8 +1,8 @@
 Checking disk quota on $HOME...
 Disk usage: X / LIMIT GiB and X / LIMIT files
 Installing VSCode extensions that are on the local machine on narval.
-(narval) $ cd $SCRATCH && cd $SCRATCH && sbatch --parsable --wckey=milatools_test --account=SLURM_ACCOUNT --nodes=1 --ntasks=1 --cpus-per-task=1 --mem=1G --time=0:05:00 --oversubscribe
---job-name=mila-code --wrap 'srun sleep 7d'
+(narval) $ cd $SCRATCH && sbatch --parsable --wckey=milatools_test --account=SLURM_ACCOUNT --nodes=1 --ntasks=1 --cpus-per-task=1 --mem=1G --time=0:05:00 --oversubscribe --job-name=mila-code --wrap
+'srun sleep 7d'
 JOB_ID
 
 ✓ Syncing vscode extensions:  N/N ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00 H:MM:SS