diff --git a/models/spleen_deepedit_annotation/configs/inference.json b/models/spleen_deepedit_annotation/configs/inference.json
index f4ff41a9..beec2b49 100644
--- a/models/spleen_deepedit_annotation/configs/inference.json
+++ b/models/spleen_deepedit_annotation/configs/inference.json
@@ -183,13 +183,6 @@
         ]
     },
     "handlers": [
-        {
-            "_target_": "CheckpointLoader",
-            "load_path": "$@bundle_root + '/models/model.pt'",
-            "load_dict": {
-                "model": "@network"
-            }
-        },
         {
             "_target_": "StatsHandler",
             "iteration_log": false
@@ -205,8 +198,16 @@
         "val_handlers": "@handlers",
         "amp": true
     },
+    "checkpointloader": {
+        "_target_": "CheckpointLoader",
+        "load_path": "$@bundle_root + '/models/model.pt'",
+        "load_dict": {
+            "model": "@network"
+        }
+    },
     "initialize": [
-        "$monai.utils.set_determinism(seed=123)"
+        "$monai.utils.set_determinism(seed=123)",
+        "$@checkpointloader(@evaluator)"
     ],
     "run": [
         "$@evaluator.run()"
diff --git a/models/spleen_deepedit_annotation/configs/inference_trt.json b/models/spleen_deepedit_annotation/configs/inference_trt.json
index 655fe010..bee0b679 100644
--- a/models/spleen_deepedit_annotation/configs/inference_trt.json
+++ b/models/spleen_deepedit_annotation/configs/inference_trt.json
@@ -5,7 +5,6 @@
         "$import ignite",
         "$import torch_tensorrt"
     ],
-    "handlers#0#_disabled_": true,
     "network_def": "$torch.jit.load(@bundle_root + '/models/model_trt.ts')",
     "evaluator#amp": false
 }
diff --git a/models/spleen_deepedit_annotation/configs/metadata.json b/models/spleen_deepedit_annotation/configs/metadata.json
index fcdd213d..115192fc 100644
--- a/models/spleen_deepedit_annotation/configs/metadata.json
+++ b/models/spleen_deepedit_annotation/configs/metadata.json
@@ -1,7 +1,8 @@
 {
     "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
-    "version": "0.5.1",
+    "version": "0.5.2",
     "changelog": {
+        "0.5.2": "update the checkpoint loader logic for inference",
         "0.5.1": "add option to validate at training start, and I/O param entries",
         "0.5.0": "enable finetune and early stop",
         "0.4.9": "fix orientation issue on clicks",