diff --git a/apax/md/nvt.py b/apax/md/nvt.py
index a98561dc..3c07dfe6 100644
--- a/apax/md/nvt.py
+++ b/apax/md/nvt.py
@@ -271,7 +271,7 @@ def body_fn(i, state):
                     )
                 ckpt = {"state": state, "step": step}
                 checkpoints.save_checkpoint(
-                    ckpt_dir=ckpt_dir,
+                    ckpt_dir=ckpt_dir.resolve(),
                     target=ckpt,
                     step=step,
                     overwrite=True,