diff --git a/importexp b/importexp
index 62d5107a1ab8e7ee0b2ecedfa29561b218863eb7..b8971022790142205f897208203959a9a93a7a35 100755
--- a/importexp
+++ b/importexp
@@ -108,7 +108,7 @@ done\
 
     sed -r '
         s@^(\s*)(cat\s+\$thisscript)@\1: \2@
-        /^\s*\$\{START\}/ i'"$SHOW_HERE_DOCS"'
+        /^\s*(\$\{START\}|srun)/ i'"$SHOW_HERE_DOCS"'
         s:^(\s*)(set\s+\+x\s*$|(set\s+\+\w+)x):\1\3:
         '"${REMOVE_ERROR_FLAG_SH-}"'
         s:^(\s*)SCRIPT_DIR=.*$:\1SCRIPT_DIR='$WORK_DIR':
@@ -116,7 +116,7 @@ done\
         s:^(\s*)BUILD_DIR=.*$:\1BUILD_DIR='$MODEL_BASE_DIR/$BUILD_SUBDIR':
         s:^(\s*)EXPDIR=.*$:\1EXPDIR='$WORK_DIR':
         s:^(\s*\.\s+)\./:\1'$RUN_BASE_DIR'/:
-        s:^(\s*(export\s+)?MODEL=")\$\{basedir\}:\1'$MODEL_BASE_DIR':
+        s:^(\s*(export\s+)?basedir=).*$:\1'$MODEL_BASE_DIR/$BUILD_SUBDIR':
     ' "$RUN_SCRIPT" > $RUN_FILE
 
     # Get script shell without possible +x/-e settings