diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py
index a4758c03d..6acc4ebad 100644
--- a/tests/handlers/test_federation.py
+++ b/tests/handlers/test_federation.py
@@ -52,7 +52,7 @@ class FederationTestCase(unittest.TestCase):
                 "get_users_in_room",
                 "bulk_get_push_rules",
                 "get_current_state",
-                "set_actions_for_event_and_users",
+                "set_push_actions_for_event_and_users",
             ]),
             resource_for_federation=NonCallableMock(),
             http_client=NonCallableMock(spec_set=[]),
diff --git a/tests/handlers/test_room.py b/tests/handlers/test_room.py
index ba20b3194..ff2b59712 100644
--- a/tests/handlers/test_room.py
+++ b/tests/handlers/test_room.py
@@ -46,7 +46,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
                 "get_users_in_room",
                 "bulk_get_push_rules",
                 "get_current_state",
-                "set_actions_for_event_and_users",
+                "set_push_actions_for_event_and_users",
             ]),
             resource_for_federation=NonCallableMock(),
             http_client=NonCallableMock(spec_set=[]),