fix test

vyperlang · Oct 4, 2024 · f8e2f17 · f8e2f17
1 parent 9d691d1
commit f8e2f17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/ast/test_annotate_and_optimize_ast.py b/tests/unit/ast/test_annotate_and_optimize_ast.py
@@ -31,7 +31,7 @@ def get_contract_info(source_code):
     pre_parse_result = pre_parse(source_code)
     py_ast = python_ast.parse(pre_parse_result.reformatted_code)
 
-    annotate_python_ast(py_ast, pre_parse_result)
+    annotate_python_ast(py_ast, pre_parse_result.reformatted_code, pre_parse_result)
 
     return py_ast, pre_parse_result.reformatted_code