diff --git a/entrypoint.sh b/entrypoint.sh
index 7deec3f..1f1dce4 100644
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -7,7 +7,7 @@ CKPT=""
 
 if compgen -G "${CKPT_DIR}/version_*/checkpoints/*.ckpt" > /dev/null; then
   CKPT=$(ls -t ${CKPT_DIR}/version_*/checkpoints/*.ckpt | head -n1)
-  RESUME_ARG="--trainer.resume_from_checkpoint $CKPT"
+  RESUME_ARG="--ckpt_path $CKPT"
 
   echo "Resuming: $CKPT"
 fi