diff --git a/src/parse.py b/src/parse.py
index 9a37535..f8970d0 100644
--- a/src/parse.py
+++ b/src/parse.py
@@ -164,7 +164,7 @@
 
 
 # Parses tokens
-def parse(tokens):
+def parse(tokens, context):
     stream = SyntaxStream(tokens)
-    cleared = NoteSkipper().clear_notes(stream, None)
+    cleared = NoteSkipper().clear_notes(stream, context)
     return cleared
diff --git a/tests/parse/test_parse.py b/tests/parse/test_parse.py
index 8b49053..2ba5517 100644
--- a/tests/parse/test_parse.py
+++ b/tests/parse/test_parse.py
@@ -97,16 +97,16 @@
 # Tests the parser wrapper works correctly
 # We expect the following behaviour:
 # - Notes to be removed from the tokens
-@given(lists(draw_token_classified()))
-def test_parse_fuzz(tokens):
+@given(lists(draw_token_classified()), draw_parse_context())
+def test_parse_fuzz(tokens, context):
     result = None
     try:
         stream = SyntaxStream(tokens.copy())
-        result = NoteSkipper().clear_notes(stream, None)
+        result = NoteSkipper().clear_notes(stream, context)
     except ParseErrorException as e:
         result = e
     try:
-        parsed = parse(tokens)
+        parsed = parse(tokens, context)
         assert parsed == result
     except ParseErrorException as e:
         assert e == result