diff --git a/tests/functional/syntax/test_as_wei_value.py b/tests/functional/syntax/test_as_wei_value.py
index 4ed3f0a224..1a225a2bef 100644
--- a/tests/functional/syntax/test_as_wei_value.py
+++ b/tests/functional/syntax/test_as_wei_value.py
@@ -6,6 +6,7 @@
     InvalidType,
     OverflowException,
     StructureException,
+    UndeclaredDefinition
 )
 
 fail_list = [
@@ -15,7 +16,7 @@
 def foo():
     x: uint256 = as_wei_value(5, szabo)
     """,
-        ArgumentException,
+        UndeclaredDefinition,
     ),
     (
         """
@@ -65,7 +66,7 @@ def foo():
         """
 FOO: constant(uint256) = as_wei_value(5, szabo)
     """,
-        ArgumentException,
+        UndeclaredDefinition,
     ),
     (
         """