diff --git a/src/sagemaker/hyperpod/inference/constant.py b/src/sagemaker/hyperpod/inference/constant.py
index edf6fa78..f2ddde56 100644
--- a/src/sagemaker/hyperpod/inference/constant.py
+++ b/src/sagemaker/hyperpod/inference/constant.py
@@ -39,13 +39,13 @@
         "mig-4g.71gb",
         "mig-7g.141gb"
     ],
-    "p6-b200.48xlarge": [
+    "ml.p6-b200.48xlarge": [
         "mig-1g.23gb",
-        "mig-1g.47gb",
-        "mig-2g.47gb",
-        "mig-3g.93gb",
-        "mig-4g.93gb",
-        "mig-7g.186gb"
+        "mig-1g.45gb",
+        "mig-2g.45gb",
+        "mig-3g.90gb",
+        "mig-4g.90gb",
+        "mig-7g.180gb"
     ],
     "ml.p6e-gb200.36xlarge": [
         "mig-1g.23gb",
diff --git a/src/sagemaker/hyperpod/training/constants.py b/src/sagemaker/hyperpod/training/constants.py
index 29f58fa8..0c51ed10 100644
--- a/src/sagemaker/hyperpod/training/constants.py
+++ b/src/sagemaker/hyperpod/training/constants.py
@@ -131,7 +131,7 @@
     'ml.p5.48xlarge': ['mig-1g.10gb', 'mig-1g.20gb', 'mig-2g.20gb', 'mig-3g.40gb', 'mig-4g.40gb', 'mig-7g.80gb'],
     'ml.p5e.48xlarge': ['mig-1g.18gb', 'mig-1g.35gb', 'mig-2g.35gb', 'mig-3g.71gb', 'mig-4g.71gb', 'mig-7g.141gb'],
     'ml.p5en.48xlarge': ['mig-1g.18gb', 'mig-1g.35gb', 'mig-2g.35gb', 'mig-3g.71gb', 'mig-4g.71gb', 'mig-7g.141gb'],
-    'p6-b200.48xlarge': ['mig-1g.23gb', 'mig-1g.45gb', 'mig-2g.45gb', 'mig-3g.90gb', 'mig-4g.90gb', 'mig-7g.180gb'],
+    'ml.p6-b200.48xlarge': ['mig-1g.23gb', 'mig-1g.45gb', 'mig-2g.45gb', 'mig-3g.90gb', 'mig-4g.90gb', 'mig-7g.180gb'],
     'ml.p6e-gb200.36xlarge': ['mig-1g.23gb', 'mig-1g.47gb', 'mig-2g.47gb', 'mig-3g.93gb', 'mig-4g.93gb', 'mig-7g.186gb'],
     'ml.g7e.2xlarge': ['mig-1g.24gb', 'mig-2g.48gb', 'mig-4g.96gb'],
     'ml.g7e.4xlarge': ['mig-1g.24gb', 'mig-2g.48gb', 'mig-4g.96gb'],
diff --git a/test/unit_tests/inference/test_hp_jumpstart_endpoint.py b/test/unit_tests/inference/test_hp_jumpstart_endpoint.py
index a418dea9..e6079eb9 100644
--- a/test/unit_tests/inference/test_hp_jumpstart_endpoint.py
+++ b/test/unit_tests/inference/test_hp_jumpstart_endpoint.py
@@ -397,7 +397,7 @@ def test_validate_mig_profile_edge_cases(self):
             ("ml.p5.48xlarge", "mig-3g.40gb"),
             ("ml.p5e.48xlarge", "mig-1g.18gb"),
             ("ml.p5en.48xlarge", "mig-7g.141gb"),
-            ("p6-b200.48xlarge", "mig-1g.23gb"),
+            ("ml.p6-b200.48xlarge", "mig-1g.23gb"),
             ("ml.p6e-gb200.36xlarge", "mig-7g.186gb"),
         ]