diff --git a/tests/unit/ast/test_annotate_and_optimize_ast.py b/tests/unit/ast/test_annotate_and_optimize_ast.py
index 7e1641e49e..67669c24c2 100644
--- a/tests/unit/ast/test_annotate_and_optimize_ast.py
+++ b/tests/unit/ast/test_annotate_and_optimize_ast.py
@@ -28,10 +28,18 @@ def foo() -> int128:
 
 
 def get_contract_info(source_code):
-    _, loop_var_annotations, class_types, reformatted_code = pre_parse(source_code)
+    (
+        _,
+        loop_var_annotations,
+        native_hex_literal_locations,
+        class_types,
+        reformatted_code,
+    ) = pre_parse(source_code)
     py_ast = python_ast.parse(reformatted_code)
 
-    annotate_python_ast(py_ast, reformatted_code, loop_var_annotations, class_types)
+    annotate_python_ast(
+        py_ast, reformatted_code, loop_var_annotations, native_hex_literal_locations, class_types
+    )
 
     return py_ast, reformatted_code