diff --git a/stdout-flush-wrapper-debug.sh b/stdout-flush-wrapper-debug.sh
index 2588eca..abb6c48 100755
--- a/stdout-flush-wrapper-debug.sh
+++ b/stdout-flush-wrapper-debug.sh
@@ -6,7 +6,9 @@
 # but not completely, so we just give LAVA more time with some sleep after
 # command termination.
 
+delay="$1"
+shift
 cmd="$1"
 shift
 stdbuf -oL "$cmd" "$@"
-sleep 10
+sleep $delay