diff --git a/sagemaker-serve/tests/integ/test_model_customization_deployment.py b/sagemaker-serve/tests/integ/test_model_customization_deployment.py
index b98282d12d..615bb67d2c 100644
--- a/sagemaker-serve/tests/integ/test_model_customization_deployment.py
+++ b/sagemaker-serve/tests/integ/test_model_customization_deployment.py
@@ -255,7 +255,7 @@ def test_sft_trainer_build(self, training_job_name):
             model="meta-textgeneration-llama-3-2-1b-instruct",
             training_dataset="s3://dummy/data.jsonl",
             accept_eula=True,
-            model_package_group_name="test-group"
+            model_package_group="test-group"
         )
         trainer._latest_training_job = training_job
 
@@ -282,7 +282,7 @@ def test_dpo_trainer_build(self, training_job_name):
                 model="meta-textgeneration-llama-3-2-1b-instruct",
                 training_dataset="s3://dummy/data.jsonl",
                 accept_eula=True,
-                model_package_group_name="test-group"
+                model_package_group="test-group"
             )
         trainer._latest_training_job = training_job