diff --git a/tests/storage/test_event_push_actions.py b/tests/storage/test_event_push_actions.py
index cc6d6c99d058..9af3891d0ffd 100644
--- a/tests/storage/test_event_push_actions.py
+++ b/tests/storage/test_event_push_actions.py
@@ -182,7 +182,7 @@ def _assert_counts(notif_count: int, highlight_count: int) -> None:
                     user_id,
                 )
             )
-            self.assertEqual(sum(aggregate_counts.values()), notif_count)
+            self.assertEqual(aggregate_counts[room_id], notif_count)
 
         def _create_event(highlight: bool = False) -> str:
             result = self.helper.send_event(