diff --git a/tests/functional/codegen/features/iteration/test_for_in_list.py b/tests/functional/codegen/features/iteration/test_for_in_list.py
index fb01cc98eb..b634735d31 100644
--- a/tests/functional/codegen/features/iteration/test_for_in_list.py
+++ b/tests/functional/codegen/features/iteration/test_for_in_list.py
@@ -772,7 +772,7 @@ def test_for() -> int128:
         a = i
     return a
     """,
-        TypeMismatch,
+        InvalidType,
     ),
     (
         """