diff --git a/tests/functional/flakes/flakes.sh b/tests/functional/flakes/flakes.sh
index c4b18a21b..e7dffde07 100644
--- a/tests/functional/flakes/flakes.sh
+++ b/tests/functional/flakes/flakes.sh
@@ -71,8 +71,9 @@ cat > "$nonFlakeDir/shebang.sh" <<EOF
 #! nix --offline shell
 #! nix flake1#fooScript
 #! nix --no-write-lock-file --command bash
-set -e
+set -ex
 foo
+echo "\$@"
 EOF
 chmod +x "$nonFlakeDir/shebang.sh"
 
@@ -524,3 +525,4 @@ expectStderr 1 nix flake metadata "$flake2Dir" --no-allow-dirty --reference-lock
 
 # Test shebang
 [[ $($nonFlakeDir/shebang.sh) = "foo" ]]
+[[ $($nonFlakeDir/shebang.sh "bar") = "foo"$'\n'"bar" ]]