diff --git a/tests/Vertex_test.py b/tests/Vertex_test.py
index ae912c7..58a0d56 100644
--- a/tests/Vertex_test.py
+++ b/tests/Vertex_test.py
@@ -1,4 +1,5 @@
 import unittest
+
 from defigraph.Vertex import Vertex
 import pytest
 from web3 import Web3
@@ -26,12 +27,14 @@ def setUp(self):
         }
 
     def test_address_should_be_valid_checksum(self):
+        name = self.pool["token0"]
+        decimals = int(self.pool["token0"]["decimals"])
+        address = self.pool["token0"]["id"]  # should be checksum address
         with pytest.raises(ValueError):
-            name = self.pool["token0"]
-            decimals = int(self.pool["token0"]["decimals"])
-            address = self.pool["token0"]["id"]
             Vertex(name, decimals, address)
 
+        Vertex(name, decimals, Web3.to_checksum_address(address))
+
     def tearDown(self):
         return super().tearDown()