diff --git a/checkpoint/orbax/checkpoint/_src/testing/benchmarks/p2p_checkpoint_manager_benchmark_test.py b/checkpoint/orbax/checkpoint/_src/testing/benchmarks/p2p_checkpoint_manager_benchmark_test.py
index d13cb28f2..6000a393c 100644
--- a/checkpoint/orbax/checkpoint/_src/testing/benchmarks/p2p_checkpoint_manager_benchmark_test.py
+++ b/checkpoint/orbax/checkpoint/_src/testing/benchmarks/p2p_checkpoint_manager_benchmark_test.py
@@ -317,12 +317,12 @@ def test_test_fn_applies_benchmark_options_correctly(self, options):
 
 class HelperFunctionsTest(parameterized.TestCase):
 
-  @mock.patch.object(p2p_checkpoint_manager, 'CheckpointManager', autospec=True)
   @mock.patch.object(p2p_options, 'CheckpointManagerOptions', autospec=True)
+  @mock.patch.object(p2p_checkpoint_manager, 'CheckpointManager', autospec=True)
   def test_create_checkpoint_manager(
       self,
-      mock_checkpoint_manager_options_cls,
       mock_checkpoint_manager_cls,
+      mock_checkpoint_manager_options_cls,
   ):
     local_dir = epath.Path('/tmp/local')
     persistent_dir = epath.Path('/tmp/persistent')