diff --git a/models/maisi_ct_generative/configs/metadata.json b/models/maisi_ct_generative/configs/metadata.json
index d6cc5330..2838ca92 100644
--- a/models/maisi_ct_generative/configs/metadata.json
+++ b/models/maisi_ct_generative/configs/metadata.json
@@ -1,7 +1,8 @@
 {
     "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_generator_ldm_20240318.json",
-    "version": "0.4.1",
+    "version": "0.4.2",
     "changelog": {
+        "0.4.2": "update train.json to fix finetune ckpt bug",
         "0.4.1": "update large files",
         "0.4.0": "update to use monai 1.4, model ckpt updated, rm GenerativeAI repo, add quality check",
         "0.3.6": "first oss version"
diff --git a/models/maisi_ct_generative/configs/train.json b/models/maisi_ct_generative/configs/train.json
index ab88ffad..f2c91fab 100644
--- a/models/maisi_ct_generative/configs/train.json
+++ b/models/maisi_ct_generative/configs/train.json
@@ -104,7 +104,7 @@
     "copy_controlnet_state": "$monai.networks.utils.copy_model_state(@controlnet, @diffusion_unet.state_dict())",
     "checkpoint_controlnet": "$torch.load(@trained_controlnet_path)",
     "load_controlnet": "$@controlnet.load_state_dict(@checkpoint_controlnet['controlnet_state_dict'], strict=True)",
-    "scale_factor": "$@checkpoint_controlnet['scale_factor'].to(@device)",
+    "scale_factor": "$@checkpoint_diffusion_unet['scale_factor'].to(@device)",
     "loss": {
         "_target_": "torch.nn.L1Loss",
         "reduction": "none"
@@ -214,7 +214,8 @@
                 "_target_": "CheckpointSaver",
                 "save_dir": "@ckpt_dir",
                 "save_dict": {
-                    "controlnet_state_dict": "@controlnet"
+                    "controlnet_state_dict": "@controlnet",
+                    "optimizer": "@optimizer"
                 },
                 "save_interval": 1,
                 "n_saved": 5