diff --git a/test/typed/statements/test_if.rb b/test/typed/statements/test_if.rb
index 32441c9a..93535f74 100644
--- a/test/typed/statements/test_if.rb
+++ b/test/typed/statements/test_if.rb
@@ -1,21 +1,23 @@
 require_relative 'helper'
 
 module Register
-class TestIfStatement #< MiniTest::Test
+class TestIfStatement < MiniTest::Test
   include Statements
 
   def test_if_basicr
-    @input = <<HERE
-class Space
-  int main()
-    if_plus( 10 - 12)
-      return 3
-    else
-      return 4
-    end
-  end
-end
-HERE
+    @input =         s(:statements,
+          s(:if_statement, :plus,
+            s(:condition,
+              s(:operator_value, :-,
+                s(:int, 10),
+                s(:int, 12))),
+            s(:true_statements,
+              s(:return,
+                s(:int, 3))),
+            s(:false_statements,
+              s(:return,
+                s(:int, 4)))))
+
   @expect =  [Label, LoadConstant,LoadConstant, OperatorInstruction,IsPlus ,
                 LoadConstant,SetSlot,Branch , Label , LoadConstant ,SetSlot,
                 Label,Label,FunctionReturn]
@@ -24,15 +26,17 @@ HERE
 
 
   def test_if_small_minus
-    @input = <<HERE
-class Space
-  int main()
-    if_minus( 10 - 12)
-      return 3
-    end
-  end
-end
-HERE
+    @input =         s(:statements,
+          s(:if_statement, :minus,
+            s(:condition,
+              s(:operator_value, :-,
+                s(:int, 10),
+                s(:int, 12))),
+            s(:true_statements,
+              s(:return,
+                s(:int, 3))),
+            s(:false_statements, nil)))
+
   @expect =  [Label, LoadConstant, LoadConstant, OperatorInstruction, IsMinus, Branch, Label ,
                LoadConstant, SetSlot, Label, Label, FunctionReturn]
   check
@@ -40,15 +44,17 @@ HERE
 
 
   def test_if_small_zero
-    @input = <<HERE
-class Space
-  int main()
-    if_zero( 10 - 12)
-      return 3
-    end
-  end
-end
-HERE
+    @input =   s(:statements,
+          s(:if_statement, :zero,
+            s(:condition,
+              s(:operator_value, :-,
+                s(:int, 10),
+                s(:int, 12))),
+            s(:true_statements,
+              s(:return,
+                s(:int, 3))),
+            s(:false_statements, nil)))
+
   @expect =  [Label, LoadConstant,LoadConstant,OperatorInstruction,IsZero ,
                 Branch , Label , LoadConstant ,SetSlot,
                 Label,Label, FunctionReturn]