diff --git a/tests/test_client.py b/tests/test_client.py
index d72b8d8..5fe3352 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -4,11 +4,16 @@
 import pytest
 import requests_mock
 
-from leakix import Client
-from leakix.client import Scope
-from leakix.field import CountryField, PluginField, PortField
-from leakix.plugin import Plugin
-from leakix.query import MustQuery, RawQuery
+from leakix import (
+    Client,
+    CountryField,
+    MustQuery,
+    Plugin,
+    PluginField,
+    PortField,
+    RawQuery,
+    Scope,
+)
 
 RESULTS_DIR = Path(__file__).parent / "results"
 HOSTS_RESULTS_DIR = RESULTS_DIR / "host"
diff --git a/tests/test_query.py b/tests/test_query.py
index 7297d77..ae6e8a3 100644
--- a/tests/test_query.py
+++ b/tests/test_query.py
@@ -2,24 +2,22 @@
 
 import pytest
 
-from leakix.field import (
+from leakix import (
     AgeField,
     CountryField,
     CustomField,
+    EmptyQuery,
     IPField,
+    MustNotQuery,
+    MustQuery,
     Operator,
+    Plugin,
     PluginField,
     PortField,
-    TimeField,
-    UpdateDateField,
-)
-from leakix.plugin import Plugin
-from leakix.query import (
-    EmptyQuery,
-    MustNotQuery,
-    MustQuery,
     RawQuery,
     ShouldQuery,
+    TimeField,
+    UpdateDateField,
 )
 
 
diff --git a/tests/test_response.py b/tests/test_response.py
index fef37f3..9fb34ea 100644
--- a/tests/test_response.py
+++ b/tests/test_response.py
@@ -1,6 +1,6 @@
 from unittest.mock import Mock
 
-from leakix.response import (
+from leakix import (
     ErrorResponse,
     RateLimitResponse,
     SuccessResponse,