diff --git a/test/test_go.rb b/test/test_go.rb
index d252c00..ff4eb1d 100644
--- a/test/test_go.rb
+++ b/test/test_go.rb
@@ -77,8 +77,10 @@ class Tmux
   end
 
   def close
-    send_keys 'C-c', 'C-u', 'exit', :Enter
-    wait { closed? }
+    wait do
+      send_keys 'C-c', 'C-u', 'exit', :Enter
+      closed?
+    end
   end
 
   def kill