diff --git a/tests/unit/lbrynet_daemon/test_Daemon.py b/tests/unit/lbrynet_daemon/test_Daemon.py index 0f0dcef3e..37f2c2ce4 100644 --- a/tests/unit/lbrynet_daemon/test_Daemon.py +++ b/tests/unit/lbrynet_daemon/test_Daemon.py @@ -1,5 +1,5 @@ import unittest -import mock +from unittest import mock import json from lbrynet.conf import Config diff --git a/tests/unit/stream/test_downloader.py b/tests/unit/stream/test_downloader.py index cd791dfd3..c89f60fdb 100644 --- a/tests/unit/stream/test_downloader.py +++ b/tests/unit/stream/test_downloader.py @@ -1,5 +1,5 @@ import os -import mock +from unittest import mock import asyncio import contextlib from lbrynet.stream.descriptor import StreamDescriptor