diff --git a/tests/test_parse.py b/tests/test_parse.py
index 83046d7..eb9e366 100644
--- a/tests/test_parse.py
+++ b/tests/test_parse.py
@@ -12,10 +12,10 @@
 from tests import test_tokenize
 
 
-# Draws random tokens and a list without whitespace in them
+# Draws classified tokens and a list without whitespace in them
 @composite
 def draw_tokens_whitespace(draw):
-    input = draw(lists(test_tokenize.draw_token_random()))
+    input = draw(lists(test_tokenize.draw_token_classified()))
     tokens = []
     for t in input:
         if t.type not in [tokenize.TokenType.SPACE, tokenize.TokenType.NEWLINE]:
@@ -36,7 +36,7 @@
 # Tests the parser wrapper works correctly
 # We expect the following behaviour:
 # - Whitespace tokens are stripped
-@given(lists(test_tokenize.draw_token_random()))
+@given(lists(test_tokenize.draw_token_classified()))
 def test_parse_fuzz(tokens):
     stripped = parse.strip_whitespace(tokens)
     parsed = parse.parse(tokens)