diff --git a/sagemaker-train/tests/integ/jumpstart/test_jumpstart_train.py b/sagemaker-train/tests/integ/jumpstart/test_jumpstart_train.py
index 2ee33f760c..298ea85e3e 100644
--- a/sagemaker-train/tests/integ/jumpstart/test_jumpstart_train.py
+++ b/sagemaker-train/tests/integ/jumpstart/test_jumpstart_train.py
@@ -30,7 +30,7 @@
             },
             # Override default instance type; the model's default
             # (ml.p3.2xlarge) is deprecated.
-            "compute": Compute(instance_type="ml.g5.xlarge"),
+            "compute": Compute(instance_type="ml.g4dn.xlarge"),
         },
         {"model_id": "xgboost-classification-model"},
         {"model_id": "catboost-regression-model"},
diff --git a/sagemaker-train/tests/integ/train/test_llm_as_judge_base_model_fix.py b/sagemaker-train/tests/integ/train/test_llm_as_judge_base_model_fix.py
index dcef7d4881..7490a373b5 100644
--- a/sagemaker-train/tests/integ/train/test_llm_as_judge_base_model_fix.py
+++ b/sagemaker-train/tests/integ/train/test_llm_as_judge_base_model_fix.py
@@ -76,6 +76,7 @@
 }
 
 
+@pytest.mark.serial
 class TestLLMAsJudgeBaseModelFix:
     """Integration test for base model fix in LLMAsJudgeEvaluator"""