diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py
index 548bae7a8df59eff4b0790c9f7aa6b806f2ce6a1..98cfbe50b3668d2b1527e44d70045c90dfc972d9 100644
--- a/tests/handlers/test_federation.py
+++ b/tests/handlers/test_federation.py
@@ -98,7 +98,7 @@ class FederationTestCase(unittest.TestCase):
         yield self.handlers.federation_handler.on_receive_pdu(pdu, False)
 
         self.datastore.persist_event.assert_called_once_with(
-            ANY, is_new_state=False, backfilled=False
+            ANY, is_new_state=False, backfilled=False, current_state=None
         )
 
         self.state_handler.annotate_event_with_state.assert_called_once_with(