diff --git a/test/corpus/assert.txt b/test/corpus/assert.txt
index fb84bdd..8b8df21 100644
--- a/test/corpus/assert.txt
+++ b/test/corpus/assert.txt
@@ -1,5 +1,5 @@
 ================================================================================
-assert statements with error message
+assert logic with error message
 ================================================================================
 
 assert 1==2 , "Wrong"
@@ -17,22 +17,23 @@ assert 1==2 , "Wrong"
           (string_end))))
 
 ================================================================================
-assert statements with 'if' condition
+assert logic with 'if' condition
 ================================================================================
 
-assert 1 == 2 if True, "error message"
+assert 1 == 2 if 3 > 2, "error message"
 
 --------------------------------------------------------------------------------
 
 (module
-  (if_statement
-    (true)
-    (assert_statement
-      (comparison_operator
-        (integer)
-        (integer))
-      (string
-        (string_start)
-        (string_content)
-        (string_end)))
-  ))
\ No newline at end of file
+      (assert_statement
+        (comparison_operator
+          (integer)
+          (integer))
+        (comparison_operator
+          (integer)
+          (integer))
+        (string
+          (string_start)
+          (string_content)
+          (string_end))))
+
diff --git a/test/corpus/expr.txt b/test/corpus/expr.txt
index d5829f2..5b2c2ea 100644
--- a/test/corpus/expr.txt
+++ b/test/corpus/expr.txt
@@ -84,4 +84,22 @@ print("hello world", end="")
           (string_start)
           (string_end))))))
 
+================================================================================
+Comprehensions
+================================================================================
+
+[x * x for x in range(5)] 
 
+--------------------------------------------------------------------------------
+
+(module
+      (list_comprehension
+        (binary_operator
+          (identifier)
+          (identifier))
+        (for_in_clause
+          (identifier)
+          (call
+            (identifier)
+            (argument_list
+              (integer))))))