diff --git a/.travis.yml b/.travis.yml
index d27f5629b..69685b8ad 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -21,7 +21,7 @@ jobs:
       install:
         - pip install coverage
         - pip install git+https://github.com/lbryio/torba.git#egg=torba
-        - pip install -e .[test]
+        - pip install -e .
       script:
         - HOME=/tmp coverage run -p --source=lbrynet -m unittest discover -vv tests.unit
       after_success:
diff --git a/tests/dht_mocks.py b/tests/dht_mocks.py
index 46dc82668..3caa8eef7 100644
--- a/tests/dht_mocks.py
+++ b/tests/dht_mocks.py
@@ -1,7 +1,7 @@
 import typing
 import contextlib
 import socket
-import mock
+from unittest import mock
 import functools
 import asyncio
 if typing.TYPE_CHECKING: