From f8e2f17a73da49b05af3ca20ba01e9df2d9fa48c Mon Sep 17 00:00:00 2001
From: tserg <8017125+tserg@users.noreply.github.com>
Date: Fri, 4 Oct 2024 22:22:48 +0800
Subject: [PATCH] fix test

---
 tests/unit/ast/test_annotate_and_optimize_ast.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/ast/test_annotate_and_optimize_ast.py b/tests/unit/ast/test_annotate_and_optimize_ast.py
index 936c81e21a..39ea899bd9 100644
--- 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