fix another test

vyperlang · Oct 4, 2024 · c25cac2 · c25cac2
1 parent 0d4edee
commit c25cac2
Showing 1 changed file with 10 additions and 2 deletions.
diff --git 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