1 files changed, 2 insertions, 2 deletions
diff --git a/tests/rest/client/test_identity.py b/tests/rest/client/test_identity.py
index 4224b0a92e..3d81618f19 100644
--- a/tests/rest/client/test_identity.py
+++ b/tests/rest/client/test_identity.py
@@ -120,7 +120,7 @@ class IdentityEnabledTestCase(unittest.HomeserverTestCase):
         # TODO: This class does not use a singleton to get it's http client
         # This should be fixed for easier testing
         # https://github.com/matrix-org/synapse-dinsic/issues/26
-        self.hs.get_handlers().identity_handler.http_client = mock_http_client
+        self.hs.get_identity_handler().http_client = mock_http_client
 
         return self.hs
 
@@ -156,7 +156,7 @@ class IdentityEnabledTestCase(unittest.HomeserverTestCase):
         )
         self.render(request)
 
-        get_json = self.hs.get_handlers().identity_handler.http_client.get_json
+        get_json = self.hs.get_identity_handler().http_client.get_json
         get_json.assert_called_once_with(
             "https://testis/_matrix/identity/api/v1/lookup",
             {"address": "test@example.com", "medium": "email"},