From b0a295da565dc2f079943d51e20bfc7554476f62 Mon Sep 17 00:00:00 2001
From: tserg <8017125+tserg@users.noreply.github.com>
Date: Tue, 26 Dec 2023 17:06:12 +0800
Subject: [PATCH] fix as wei value test

---
 tests/functional/syntax/test_as_wei_value.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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,
     ),
     (
         """