diff --git a/lib/cli/kit/support/test_helper.rb b/lib/cli/kit/support/test_helper.rb
index f4a7757..e229f6a 100644
--- a/lib/cli/kit/support/test_helper.rb
+++ b/lib/cli/kit/support/test_helper.rb
@@ -59,7 +59,7 @@ module Kit
             module System
               class << self
                 alias_method :original_system, :system
-                def system(cmd, *a, sudo: false, env: {}, stdin: nil, **kwargs)
+                def system(cmd, *a, sudo: false, env: {}, stdin: nil, **kwargs, &block)
                   a.unshift(cmd)
                   expected_command = expected_command(a, sudo: sudo, env: env)
 
@@ -69,7 +69,7 @@ def system(cmd, *a, sudo: false, env: {}, stdin: nil, **kwargs)
                   # Otherwise handle the command
                   if expected_command[:allow]
                     uself = self #: as untyped
-                    uself.original_system(*a, sudo: sudo, env: env, **kwargs)
+                    uself.original_system(*a, sudo: sudo, env: env, **kwargs, &block)
                   else
                     FakeSuccess.new(expected_command[:success])
                   end