diff --git a/test/irb/test_ruby_lex.rb b/test/irb/test_ruby_lex.rb
index 943cce33b..6a9a5963a 100644
--- a/test/irb/test_ruby_lex.rb
+++ b/test/irb/test_ruby_lex.rb
@@ -41,7 +41,6 @@ def calculate_indenting(lines, add_new_line)
       ruby_lex = RubyLex.new(context)
       mock_io = MockIO_AutoIndent.new(lines, last_line_index, byte_pointer, add_new_line)
 
-      ruby_lex.set_input { @io.gets }
       ruby_lex.configure_io(mock_io)
       mock_io.calculated_indent
     end
@@ -653,7 +652,7 @@ def assert_dynamic_prompt(lines, expected_prompt_list)
       ruby_lex.set_prompt do |ltype, indent, continue, line_no|
         '%03d:%01d:%1s:%s ' % [line_no, indent, ltype, continue ? '*' : '>']
       end
-      ruby_lex.set_input { @io.gets }
+      ruby_lex.configure_io(io)
     end
 
     def test_dyanmic_prompt