diff --git a/tests/abstract_attract_test.py b/tests/abstract_attract_test.py
index 09d0349..13d3327 100644
--- a/tests/abstract_attract_test.py
+++ b/tests/abstract_attract_test.py
@@ -1,12 +1,13 @@
-from pillar_server.tests import TestPillarServer, AbstractPillarTest
+from pillar.tests import PillarTestServer, AbstractPillarTest
 from attract_server import AttractExtension
 
-class TestAttractServer(TestPillarServer):
+
+class AttractTestServer(PillarTestServer):
     def __init__(self, *args, **kwargs):
-        super(TestAttractServer, self).__init__(*args, **kwargs)
+        super(AttractTestServer, self).__init__(*args, **kwargs)
 
         self.load_extension(AttractExtension(), '/attract')
 
 
 class AbstractAttractTest(AbstractPillarTest):
-    pillar_server_class = TestAttractServer
+    pillar_server_class = AttractTestServer