From b01887c7ec6389651234bc5f64b7ed94ea843385 Mon Sep 17 00:00:00 2001
From: Lex Berezhny <lex@damoti.com>
Date: Tue, 31 Dec 2019 15:45:23 -0500
Subject: [PATCH] fix two unit tests

---
 lbry/tests/unit/test_cli.py  | 2 +-
 lbry/tests/unit/test_conf.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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