diff --git a/horovod/spark/lightning/remote.py b/horovod/spark/lightning/remote.py
index 35800b79ae..7580b38e36 100644
--- a/horovod/spark/lightning/remote.py
+++ b/horovod/spark/lightning/remote.py
@@ -132,7 +132,7 @@ def train(serialized_model):
             # Otherwise we are seeing hanging in training.
             for cb in callbacks:
                 if isinstance(cb, ModelCheckpoint):
-                    cb.dir_path = ckpt_dir
+                    cb.dirpath = ckpt_dir
                     cb.filename = ckpt_filename
                     _checkpoint_callback = cb
                     require_checkpoint = True