diff --git a/pytorch-ddp-example/activate.sh b/pytorch-ddp-example/activate.sh
index b6f4c293b52e90f4db63de8db81ddbdbb580a27f..b5d1de750ad633d3d283a2eaa8ffa0fee46903ed 100644
--- a/pytorch-ddp-example/activate.sh
+++ b/pytorch-ddp-example/activate.sh
@@ -12,19 +12,19 @@ venv_dir="$curr_dir"/env-"$SYSTEMNAME"
 module --force purge
 if ! [ -f "$curr_dir"/modules.sh ]; then
     echo "Cannot find \`$curr_dir/modules.sh\`; its existence is required."
-    exit 1
+    return 1
 fi
 source "$curr_dir"/modules.sh
 
 if ! [ -d "$venv_dir" ]; then
     echo "Please manually execute \`nice bash ${curr_dir@Q}/set_up.sh\`" \
          "on a login node to create the \`venv\`."
-    exit 1
+    return 1
 elif ! [ -f "$venv_dir"/bin/activate ]; then
     echo "Something seems to be wrong with the \`venv\` at \`$venv_dir\`." \
          "Please delete it (\`nice rm -rf ${venv_dir@Q}\`) and" \
          "execute \`nice bash ${curr_dir@Q}/set_up.sh\`" \
          "on a login node to re-create the \`venv\`."
-    exit 1
+    return 1
 fi
 source "$venv_dir"/bin/activate
diff --git a/pytorch-ddp-example/set_up.sh b/pytorch-ddp-example/set_up.sh
index 0e45918ebeac45678b0be7f350e9d30773819ed2..2780b72196ba4c095286027798b860e46b612e95 100644
--- a/pytorch-ddp-example/set_up.sh
+++ b/pytorch-ddp-example/set_up.sh
@@ -11,7 +11,7 @@ if ! [ -d "$venv_dir" ]; then
     module --force purge
     if ! [ -f "$curr_dir"/modules.sh ]; then
         echo "Cannot find \`$curr_dir/modules.sh\`; its existence is required."
-        exit 1
+        return 1
     fi
     source "$curr_dir"/modules.sh
 
@@ -22,7 +22,7 @@ if ! [ -d "$venv_dir" ]; then
              "Please delete it (\`nice rm -rf ${venv_dir@Q}\`) and" \
              "execute \`nice bash ${curr_dir@Q}/set_up.sh\`" \
              "on a login node to re-create the \`venv\`."
-        exit 1
+        return 1
     fi
     source "$venv_dir"/bin/activate
 
@@ -30,7 +30,7 @@ if ! [ -d "$venv_dir" ]; then
     if ! [ -f "$curr_dir"/requirements.txt ]; then
         echo "Cannot find \`$curr_dir/requirements.txt\`;" \
              "its existence is required."
-        exit 1
+        return 1
     fi
     python -m pip install -r "$curr_dir"/requirements.txt
 
diff --git a/pytorch-fsdp-example/activate.sh b/pytorch-fsdp-example/activate.sh
index b6f4c293b52e90f4db63de8db81ddbdbb580a27f..b5d1de750ad633d3d283a2eaa8ffa0fee46903ed 100644
--- a/pytorch-fsdp-example/activate.sh
+++ b/pytorch-fsdp-example/activate.sh
@@ -12,19 +12,19 @@ venv_dir="$curr_dir"/env-"$SYSTEMNAME"
 module --force purge
 if ! [ -f "$curr_dir"/modules.sh ]; then
     echo "Cannot find \`$curr_dir/modules.sh\`; its existence is required."
-    exit 1
+    return 1
 fi
 source "$curr_dir"/modules.sh
 
 if ! [ -d "$venv_dir" ]; then
     echo "Please manually execute \`nice bash ${curr_dir@Q}/set_up.sh\`" \
          "on a login node to create the \`venv\`."
-    exit 1
+    return 1
 elif ! [ -f "$venv_dir"/bin/activate ]; then
     echo "Something seems to be wrong with the \`venv\` at \`$venv_dir\`." \
          "Please delete it (\`nice rm -rf ${venv_dir@Q}\`) and" \
          "execute \`nice bash ${curr_dir@Q}/set_up.sh\`" \
          "on a login node to re-create the \`venv\`."
-    exit 1
+    return 1
 fi
 source "$venv_dir"/bin/activate
diff --git a/pytorch-fsdp-example/set_up.sh b/pytorch-fsdp-example/set_up.sh
index 0e45918ebeac45678b0be7f350e9d30773819ed2..2780b72196ba4c095286027798b860e46b612e95 100644
--- a/pytorch-fsdp-example/set_up.sh
+++ b/pytorch-fsdp-example/set_up.sh
@@ -11,7 +11,7 @@ if ! [ -d "$venv_dir" ]; then
     module --force purge
     if ! [ -f "$curr_dir"/modules.sh ]; then
         echo "Cannot find \`$curr_dir/modules.sh\`; its existence is required."
-        exit 1
+        return 1
     fi
     source "$curr_dir"/modules.sh
 
@@ -22,7 +22,7 @@ if ! [ -d "$venv_dir" ]; then
              "Please delete it (\`nice rm -rf ${venv_dir@Q}\`) and" \
              "execute \`nice bash ${curr_dir@Q}/set_up.sh\`" \
              "on a login node to re-create the \`venv\`."
-        exit 1
+        return 1
     fi
     source "$venv_dir"/bin/activate
 
@@ -30,7 +30,7 @@ if ! [ -d "$venv_dir" ]; then
     if ! [ -f "$curr_dir"/requirements.txt ]; then
         echo "Cannot find \`$curr_dir/requirements.txt\`;" \
              "its existence is required."
-        exit 1
+        return 1
     fi
     python -m pip install -r "$curr_dir"/requirements.txt