diff --git a/test/irb/test_context.rb b/test/irb/test_context.rb
index c55de4f84e..d58477c84e 100644
--- a/test/irb/test_context.rb
+++ b/test/irb/test_context.rb
@@ -143,6 +143,7 @@ module TestIRB
         "_\n"
       ])
       irb = IRB::Irb.new(IRB::WorkSpace.new(Object.new), input)
+      irb.context.return_format = "=> %s\n"
 
       # The default
       irb.context.echo = true