diff --git a/lbry/tests/unit/test_cli.py b/lbry/tests/unit/test_cli.py
index b7fbdc0de..a50ee1650 100644
--- a/lbry/tests/unit/test_cli.py
+++ b/lbry/tests/unit/test_cli.py
@@ -10,7 +10,7 @@ from types import SimpleNamespace
 from contextlib import asynccontextmanager
 
 import docopt
-from torba.testcase import AsyncioTestCase
+from lbry.testcase import AsyncioTestCase
 
 from lbry.extras.cli import normalize_value, main, setup_logging
 from lbry.extras.system_info import get_platform
diff --git a/lbry/tests/unit/test_conf.py b/lbry/tests/unit/test_conf.py
index 893ce3211..cae6e6a90 100644
--- a/lbry/tests/unit/test_conf.py
+++ b/lbry/tests/unit/test_conf.py
@@ -4,6 +4,7 @@ import types
 import tempfile
 import unittest
 import argparse
+import lbry.wallet
 from lbry.conf import Config, BaseConfig, String, Integer, Toggle, Servers, Strings, StringChoice, NOT_SET
 from lbry.error import InvalidCurrencyError