diff --git a/tests/test_dump.py b/tests/test_dump.py
index 11fb316..7b5a345 100644
--- a/tests/test_dump.py
+++ b/tests/test_dump.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from mwsql.dump import Dump
+from mwsql import Dump
 
 from .helpers import Capturing
 
diff --git a/tests/test_utils.py b/tests/test_utils.py
index dbbc5f6..1d27610 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -4,7 +4,8 @@
 import pytest
 import requests
 
-from mwsql.utils import _open_file, head, load
+from mwsql import head, load
+from mwsql.utils import _open_file
 
 from .helpers import Capturing