diff --git a/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py b/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py
index 751162d2e6..d9eb39db95 100644
--- a/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py
+++ b/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py
@@ -107,7 +107,7 @@ def test_jumpstart_hub_gated_model(setup, add_model_references):
 
 def test_jumpstart_gated_model_inference_component_enabled(setup, add_model_references):
 
-    model_id = "meta-textgeneration-llama-2-7b"
+    model_id = "meta-textgeneration-llama-3-2-1b"
 
     hub_name = os.environ[ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME]