diff --git a/nix/image/scripts/integration b/nix/image/scripts/integration
index 32a9708..d8187d9 100755
--- a/nix/image/scripts/integration
+++ b/nix/image/scripts/integration
@@ -1,6 +1,6 @@
 #!/bin/nu
 
-def main [feature: path, ...rest] {
+def --wrapped main [feature: path, ...rest] {
     mut feature = $feature;
     mut workdir = $"($env.WEBROOT)/build/integration"
     if ($feature | str starts-with "apps/") {
diff --git a/nix/image/scripts/tests b/nix/image/scripts/tests
index 524172c..868f225 100755
--- a/nix/image/scripts/tests
+++ b/nix/image/scripts/tests
@@ -1,6 +1,6 @@
 #!/bin/nu
 
-def main [...rest] {
+def --wrapped main [...rest] {
     cd $env.WEBROOT
 
     XDEBUG_SESSION=haze phpunit --configuration $"($env.WEBROOT)/tests/phpunit-autotest.xml" ...$rest