diff --git a/bittensor_wallet/__init__.py b/bittensor_wallet/__init__.py
index 78d8376..b58bbfe 100644
--- a/bittensor_wallet/__init__.py
+++ b/bittensor_wallet/__init__.py
@@ -1,16 +1,16 @@
 from bittensor_wallet.bittensor_wallet import (
     # classes
-    Config,
-    Keyfile,
-    Keypair,
-    Wallet,
+    Config as Config,
+    Keyfile as Keyfile,
+    Keypair as Keypair,
+    Wallet as Wallet,
     # modules
-    config,
-    errors,
-    keyfile,
-    keypair,
-    utils,
-    wallet,
+    config as config,
+    errors as errors,
+    keyfile as keyfile,
+    keypair as keypair,
+    utils as utils,
+    wallet as wallet,
 )
 
 __version__ = "2.0.0"
diff --git a/tests/test_keyfile.py b/tests/test_keyfile.py
index 182e9f1..2663c70 100644
--- a/tests/test_keyfile.py
+++ b/tests/test_keyfile.py
@@ -23,7 +23,6 @@
 
 import pytest
 from bip39 import bip39_validate
-from jedi.common import monkeypatch
 
 from bittensor_wallet.errors import ConfigurationError, KeyFileError
 from bittensor_wallet.keyfile import Keyfile