diff --git a/tests/test_tokenize.py b/tests/test_tokenize.py
index b68baa0..d4f57b8 100644
--- a/tests/test_tokenize.py
+++ b/tests/test_tokenize.py
@@ -208,7 +208,7 @@
 # Generates a list of tokens and their corrected locations
 @composite
 def draw_tokens_locations(draw):
-    tokens = draw(lists(draw_token_random()))
+    tokens = draw(lists(draw_token_classified()))
     filename = draw(text())
     located = []
     line = 1