diff --git a/tests/rest/client/test_identity.py b/tests/rest/client/test_identity.py

index ec260a2873..36bb19a24c 100644
--- a/tests/rest/client/test_identity.py
+++ b/tests/rest/client/test_identity.py
@@ -19,7 +19,8 @@ from mock import Mock
 
 from twisted.internet import defer
 
-from synapse.rest.client.v1 import admin, login, room
+import synapse.rest.admin
+from synapse.rest.client.v1 import login, room
 from synapse.rest.client.v2_alpha import account
 
 from tests import unittest
@@ -30,7 +31,7 @@ class IdentityDisabledTestCase(unittest.HomeserverTestCase):
 
     servlets = [
         account.register_servlets,
-        admin.register_servlets,
+        synapse.rest.admin.register_servlets_for_client_rest_resource,
         room.register_servlets,
         login.register_servlets,
     ]
@@ -109,7 +110,7 @@ class IdentityEnabledTestCase(unittest.HomeserverTestCase):
 
     servlets = [
         account.register_servlets,
-        admin.register_servlets,
+        synapse.rest.admin.register_servlets_for_client_rest_resource,
         room.register_servlets,
         login.register_servlets,
     ]
@@ -154,9 +155,7 @@ class IdentityEnabledTestCase(unittest.HomeserverTestCase):
             "address": "test@example.com",
         }
         request_data = json.dumps(params)
-        request_url = (
-            "/rooms/%s/invite" % (room_id)
-        ).encode('ascii')
+        request_url = ("/rooms/%s/invite" % (room_id)).encode('ascii')
         request, channel = self.make_request(
             b"POST", request_url, request_data, access_token=self.tok,
         )