diff --git a/tests/integ/sagemaker/serve/utils/test_hardware_detector.py b/tests/integ/sagemaker/serve/utils/test_hardware_detector.py
index 9102927c55..bab26a25d1 100644
--- a/tests/integ/sagemaker/serve/utils/test_hardware_detector.py
+++ b/tests/integ/sagemaker/serve/utils/test_hardware_detector.py
@@ -19,7 +19,7 @@
 REGION = "us-west-2"
 VALID_INSTANCE_TYPE = "ml.g5.48xlarge"
 INVALID_INSTANCE_TYPE = "fl.c5.57xxlarge"
-EXPECTED_INSTANCE_GPU_INFO = (8, 196608)
+EXPECTED_INSTANCE_GPU_INFO = (8, 183104)
 
 
 def test_get_gpu_info_success(sagemaker_session):