diff --git a/dlrover/trainer/torch/elastic/checkpoint.py b/dlrover/trainer/torch/elastic/checkpoint.py
index e8858c629..bf391d135 100644
--- a/dlrover/trainer/torch/elastic/checkpoint.py
+++ b/dlrover/trainer/torch/elastic/checkpoint.py
@@ -170,7 +170,7 @@ def init_checkpoint_manager(
                 max_to_keep,
             )
         else:
-            raise ValueError(f"Not support model class {model}")
+            raise NotImplementedError(f"Not support model class {model}")
 
 
 class LocalCheckpointManger(CheckpointManger):