diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py
index cbb1eb08c..9936df52f 100644
--- a/tests/test_endpoints.py
+++ b/tests/test_endpoints.py
@@ -267,7 +267,7 @@ def square(x)
     assert response3.status_code == 200
     assert response3.json() == {
         "imports": {"errors": []},
-        "function": {"errors": ["expected ':' (<unknown>, line 4)"]},
+        "function": {"errors": ["invalid syntax (<unknown>, line 4)"]},
     }
 
     # Test case with invalid JSON payload
@@ -290,7 +290,7 @@ def square(x)
     assert response6.status_code == 200
     assert response6.json() == {
         "imports": {"errors": []},
-        "function": {"errors": ["expected ':' (<unknown>, line 4)"]},
+        "function": {"errors": ["invalid syntax (<unknown>, line 4)"]},
     }
 
 
diff --git a/tests/test_validate_code.py b/tests/test_validate_code.py
index cacf3669c..87ff6b7e4 100644
--- a/tests/test_validate_code.py
+++ b/tests/test_validate_code.py
@@ -44,7 +44,7 @@ def square(x)
     errors3 = validate_code(code3)
     assert errors3 == {
         "imports": {"errors": []},
-        "function": {"errors": ["expected ':' (<unknown>, line 4)"]},
+        "function": {"errors": ["invalid syntax (<unknown>, line 4)"]},
     }