diff --git a/test/integration_tests/inference/cli/test_cli_jumpstart_inference.py b/test/integration_tests/inference/cli/test_cli_jumpstart_inference.py
index 6d7e2c3c..5097eb0d 100644
--- a/test/integration_tests/inference/cli/test_cli_jumpstart_inference.py
+++ b/test/integration_tests/inference/cli/test_cli_jumpstart_inference.py
@@ -33,7 +33,7 @@ def test_js_create(runner, js_endpoint_name):
     result = runner.invoke(js_create, [
         "--namespace", NAMESPACE,
         "--version", VERSION,
-        "--model-id", "deepseek-llm-r1-distill-qwen-1-5b",
+        "--model-id", "huggingface-llm-phi-2",
         "--instance-type", "ml.g5.8xlarge",
         "--endpoint-name", js_endpoint_name,
     ])
diff --git a/test/integration_tests/inference/cli/test_cli_jumpstart_v12_fields.py b/test/integration_tests/inference/cli/test_cli_jumpstart_v12_fields.py
index 2ec0572a..cc9af8e1 100644
--- a/test/integration_tests/inference/cli/test_cli_jumpstart_v12_fields.py
+++ b/test/integration_tests/inference/cli/test_cli_jumpstart_v12_fields.py
@@ -33,7 +33,7 @@ def test_create_with_v12_fields(runner, endpoint_name):
     result = runner.invoke(js_create, [
         "--namespace", NAMESPACE,
         "--version", VERSION,
-        "--model-id", "deepseek-llm-r1-distill-qwen-1-5b",
+        "--model-id", "huggingface-llm-phi-2",
         "--instance-type", "ml.g5.8xlarge",
         "--endpoint-name", endpoint_name,
         "--accept-eula", "True",
diff --git a/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference.py b/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference.py
index 98bf9269..6926ebb5 100644
--- a/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference.py
+++ b/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference.py
@@ -15,7 +15,7 @@
 ENDPOINT_NAME = "js-sdk-integration-" + get_time_str()
 
 INSTANCE_TYPE = "ml.g5.8xlarge"
-MODEL_ID = "deepseek-llm-r1-distill-qwen-1-5b"
+MODEL_ID = "huggingface-llm-phi-2"
 
 TIMEOUT_MINUTES = 20
 POLL_INTERVAL_SECONDS = 30
diff --git a/test/integration_tests/inference/sdk/test_sdk_jumpstart_v12_fields.py b/test/integration_tests/inference/sdk/test_sdk_jumpstart_v12_fields.py
index 5f77f9d4..1e028608 100644
--- a/test/integration_tests/inference/sdk/test_sdk_jumpstart_v12_fields.py
+++ b/test/integration_tests/inference/sdk/test_sdk_jumpstart_v12_fields.py
@@ -22,7 +22,7 @@ def endpoint_obj():
 
     return HPJumpStartEndpoint(
         metadata=metadata,
-        model=Model(model_id="deepseek-llm-r1-distill-qwen-1-5b", accept_eula=True),
+        model=Model(model_id="huggingface-llm-phi-2", accept_eula=True),
         server=Server(instance_type="ml.g5.8xlarge"),
         sage_maker_endpoint=SageMakerEndpoint(name=ENDPOINT_NAME),
         replicas=1,
diff --git a/test/integration_tests/init/test_jumpstart_creation.py b/test/integration_tests/init/test_jumpstart_creation.py
index 04bb1f2c..cb4c02e7 100644
--- a/test/integration_tests/init/test_jumpstart_creation.py
+++ b/test/integration_tests/init/test_jumpstart_creation.py
@@ -88,7 +88,7 @@ def test_configure_jumpstart(runner, js_endpoint_name, test_directory):
         configure = init.configure
     result = runner.invoke(
         configure, [
-            "--model-id", "deepseek-llm-r1-distill-qwen-1-5b",
+            "--model-id", "huggingface-llm-phi-2",
             "--instance-type", "ml.g5.8xlarge", 
             "--endpoint-name", js_endpoint_name
         ], catch_exceptions=False
@@ -97,7 +97,7 @@ def test_configure_jumpstart(runner, js_endpoint_name, test_directory):
     
     # Verify configuration was saved correctly
     expected_config = {
-        "model_id": "deepseek-llm-r1-distill-qwen-1-5b",
+        "model_id": "huggingface-llm-phi-2",
         "instance_type": "ml.g5.8xlarge",
         "endpoint_name": js_endpoint_name
     }