diff --git a/src/mad/ruby/ssh_stream.rb b/src/mad/ruby/ssh_stream.rb
index 8d9837cb2b..b86200ef3f 100644
--- a/src/mad/ruby/ssh_stream.rb
+++ b/src/mad/ruby/ssh_stream.rb
@@ -210,7 +210,7 @@ class SshStreamCommand < RemotesCommand
         @stream.exec(command)
         @stream.stdin.write(stdin) if stdin
 
-        @code = @stream.wait_for_command
+        @code = @stream.wait_for_command(@timeout)
 
         @stdout = @stream.out
         @stderr = @stream.err