diff --git a/test_run_folder/test_run_hybrid.sh b/test_run_folder/test_run_hybrid.sh
index f5235c0..31262e1 100755
--- a/test_run_folder/test_run_hybrid.sh
+++ b/test_run_folder/test_run_hybrid.sh
@@ -15,7 +15,9 @@ if [ -f "$PROJECT_ROOT/venv/bin/python" ]; then
 else
     PYTHON="python3"
 fi
-dates=("$1")
+
+if [ $# -ge 1 ]; then
+    dates=("$1")
 else
     dates=(
         "2026-06-10" "2026-06-11" "2026-06-12" "2026-06-13" "2026-06-14"
diff --git a/test_run_folder/test_run_mortality.sh b/test_run_folder/test_run_mortality.sh
index 4b16a81..7407844 100755
--- a/test_run_folder/test_run_mortality.sh
+++ b/test_run_folder/test_run_mortality.sh
@@ -2,10 +2,16 @@
 set -euo pipefail
 
 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
-export PYTHONPATH="$SCRIPT_DIR/src"
+if [ -d "$SCRIPT_DIR/../src" ]; then
+    PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
+else
+    PROJECT_ROOT="$SCRIPT_DIR"
+fi
+cd "$PROJECT_ROOT"
+export PYTHONPATH="$PROJECT_ROOT/src"
 
-if [ -f "$SCRIPT_DIR/venv/bin/python" ]; then
-    PYTHON="$SCRIPT_DIR/venv/bin/python"
+if [ -f "$PROJECT_ROOT/venv/bin/python" ]; then
+    PYTHON="$PROJECT_ROOT/venv/bin/python"
 else
     PYTHON="python3"
 fi
diff --git a/test_run_folder/test_run_tensor_batch.sh b/test_run_folder/test_run_tensor_batch.sh
index 8d0c1b7..6130dd1 100755
--- a/test_run_folder/test_run_tensor_batch.sh
+++ b/test_run_folder/test_run_tensor_batch.sh
@@ -15,7 +15,9 @@ if [ -f "$PROJECT_ROOT/venv/bin/python" ]; then
 else
     PYTHON="python3"
 fi
-dates=("$1")
+
+if [ $# -ge 1 ]; then
+    dates=("$1")
 else
     dates=(
         "2026-06-10" "2026-06-11" "2026-06-12" "2026-06-13" "2026-06-14"