diff --git a/tests/diagnostics/test_checkpoint.py b/tests/diagnostics/test_checkpoint.py
index 030fe7d1..ed7d6259 100644
--- a/tests/diagnostics/test_checkpoint.py
+++ b/tests/diagnostics/test_checkpoint.py
@@ -15,7 +15,7 @@
 from anemoi.training.diagnostics.callbacks import AnemoiCheckpoint
 from anemoi.training.utils.jsonify import map_config_to_primitives
 
-torch.mps.set_per_process_memory_fraction(0.0)
+torch.mps.set_per_process_memory_fraction(0.7)
 
 
 class DummyModel(torch.nn.Module):