diff --git a/integration-test/test/base.py b/integration-test/test/base.py
index 947afc47..2491c4b2 100644
--- a/integration-test/test/base.py
+++ b/integration-test/test/base.py
@@ -1,9 +1,8 @@
 """An example that demonstrates low-level construction of a transaction."""
 
 import os
-import time
 
-from ogmios import OgmiosChainContext
+import ogmios as python_ogmios
 from retry import retry
 
 from pycardano import *
@@ -23,7 +22,7 @@ class TestBase:
     # TODO: Bring back kupo test
     KUPO_URL = "http://localhost:1442"
 
-    chain_context = OgmiosChainContext(
+    chain_context = python_ogmios.OgmiosChainContext(
         host="localhost", port=1337, network=Network.TESTNET
     )