diff --git a/tests/test_tokenize.py b/tests/test_tokenize.py
index 4d1be7b..52b46e5 100644
--- a/tests/test_tokenize.py
+++ b/tests/test_tokenize.py
@@ -172,6 +172,7 @@
     assert tokenize.locate_tokens(input, filename) == tokens
 
 
+# Draw a list of tokens with correct types and unknown types
 @composite
 def draw_tokens_classified(draw):
     strategies = [
@@ -186,6 +187,7 @@
     return (input, tokens)
 
 
+# Test that classification can add types properly
 @given(draw_tokens_classified())
 def test_tokenize_classification(test_data):
     (input, tokens) = test_data