diff --git a/monkey/tests/unit_tests/infection_monkey/transport/test_island_api_client.py b/monkey/tests/unit_tests/infection_monkey/transport/test_island_api_client.py
index ce9562641..513a8a3e0 100644
--- a/monkey/tests/unit_tests/infection_monkey/transport/test_island_api_client.py
+++ b/monkey/tests/unit_tests/infection_monkey/transport/test_island_api_client.py
@@ -2,8 +2,8 @@ import pytest
 import requests
 import requests_mock
 
-from infection_monkey.transport import IslandAPIClient
-from infection_monkey.transport.island_api_client_errors import (
+from infection_monkey.transport import (
+    IslandAPIClient,
     IslandAPIConnectionError,
     IslandAPIError,
     IslandAPITimeoutError,