fix tests

2 files changed, 2 insertions, 2 deletions

diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py

index a4758c03db..6acc4ebadc 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 ba20b31945..ff2b597124 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=[]),