diff --git a/test/test_go.rb b/test/test_go.rb
index cf1fd77..c09c86f 100644
--- a/test/test_go.rb
+++ b/test/test_go.rb
@@ -1229,7 +1229,7 @@ module TestShell
     tmux.send_keys :BTab, :BTab, pane: 1
     tmux.until(1) { |lines| lines[-2].include? '(2)' }
     tmux.send_keys :Enter, pane: 1
-    tmux.until { |lines| lines[-1].include? 'cat' }
+    tmux.until { |lines| lines[-1].include?('cat') || lines[-2].include?('cat') }
     tmux.send_keys :Enter
     tmux.until { |lines| lines[-1].include? 'test1test2' }
   end