diff --git a/test/reline/yamatanooroti/termination_checker.rb b/test/reline/yamatanooroti/termination_checker.rb
index 256f1bb2a2..15ec7d5913 100644
--- a/test/reline/yamatanooroti/termination_checker.rb
+++ b/test/reline/yamatanooroti/termination_checker.rb
@@ -4,11 +4,11 @@
 class TerminationChecker < RubyLex
   def terminated?(code)
     code.gsub!(/\n*$/, '').concat("\n")
-    @tokens = self.class.ripper_lex_without_warning(code)
-    continue = process_continue
-    code_block_open = check_code_block(code)
-    indent = process_nesting_level
-    ltype = process_literal_type
+    tokens = self.class.ripper_lex_without_warning(code)
+    continue = process_continue(tokens)
+    code_block_open = check_code_block(code, tokens)
+    indent = process_nesting_level(tokens)
+    ltype = process_literal_type(tokens)
     if code_block_open or ltype or continue or indent > 0
       false
     else
@@ -19,9 +19,7 @@ def terminated?(code)
 
 class AutoIndent < RubyLex
   def initialize
-    context = Struct.new("MockIRBContext", :auto_indent_mode, :workspace, :local_variables).new(true, nil, [])
-    set_input(self, context: context)
-    set_auto_indent(context)
+    @context = Struct.new("MockIRBContext", :auto_indent_mode, :workspace, :local_variables).new(true, nil, [])
   end
 
   def auto_indent(&block)