1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test_state.py b/tests/test_state.py
index 74a8ce6096..c6baea3d76 100644
--- a/tests/test_state.py
+++ b/tests/test_state.py
@@ -88,7 +88,7 @@ class _DummyStore:
 
         return groups
 
-    async def get_state_ids_for_group(self, state_group):
+    async def get_state_ids_for_group(self, state_group, state_filter=None):
         return self._group_to_state[state_group]
 
     async def store_state_group(