diff --git a/tests/framework/callbacks/test_dcp_saver.py b/tests/framework/callbacks/test_dcp_saver.py
index fc12243229..8e168c770b 100644
--- a/tests/framework/callbacks/test_dcp_saver.py
+++ b/tests/framework/callbacks/test_dcp_saver.py
@@ -991,7 +991,9 @@ class DummyStorageWriter(FileSystemWriter):
     def __init__(self, path: str) -> None:
         super().__init__(path)
 
-    def set_up_storage_writer(self, is_coordinator: bool) -> None:
+    def set_up_storage_writer(
+        self, is_coordinator: bool, *args: Any, **kwargs: Any
+    ) -> None:
         pass
 
 
@@ -999,5 +1001,7 @@ class DummyStorageReader(FileSystemReader):
     def __init__(self, path: str) -> None:
         super().__init__(path)
 
-    def set_up_storage_writer(self, is_coordinator: bool) -> None:
+    def set_up_storage_reader(
+        self, is_coordinator: bool, *args: Any, **kwargs: Any
+    ) -> None:
         pass