diff --git a/tests/persistence/test_migrations.py b/tests/persistence/test_migrations.py
index daaf72b4a..31b58068c 100644
--- a/tests/persistence/test_migrations.py
+++ b/tests/persistence/test_migrations.py
@@ -376,7 +376,7 @@ def test_migrate_set_sequence_ids():
     engine = MagicMock()
     engine.begin = MagicMock()
     engine.name = "postgresql"
-    set_sequence_ids(engine, 22, 55, 5)
+    set_sequence_ids(engine, 22, 55, 5, 3, 1)
 
     assert engine.begin.call_count == 1
     engine.reset_mock()