diff --git a/garlic/stdexp.nix b/garlic/stdexp.nix
index efddd96..52e8ab9 100644
--- a/garlic/stdexp.nix
+++ b/garlic/stdexp.nix
@@ -76,12 +76,12 @@ rec {
       inherit nextStage;
     };
 
-    srun = {nextStage, conf, ...}: (
+    srun = {nextStage, conf, preSrun ? "", ...}: (
       assert (assertMsg (!(conf ? cpuBind))
         "cpuBind is no longer available in the standard srun stage");
       stages.srun {
         inherit (conf) nixPrefix;
-        inherit nextStage;
+        inherit nextStage preSrun;
 
         # Binding is set to cores always
         cpuBind = "cores,verbose";