diff --git a/tests/unit/swarm_test.py b/tests/unit/swarm_test.py
index 55803834..39d4ec46 100644
--- a/tests/unit/swarm_test.py
+++ b/tests/unit/swarm_test.py
@@ -3,7 +3,7 @@
 import json
 
 from . import fake_api
-from ..base import requires_api_version
+from ..helpers import requires_api_version
 from .api_test import (DockerClientTest, url_prefix, fake_request)