diff --git a/olive/common/utils.py b/olive/common/utils.py
index 46a4e7207..8d25d1715 100644
--- a/olive/common/utils.py
+++ b/olive/common/utils.py
@@ -41,7 +41,7 @@ def get_package_name_from_ep(execution_provider):
         "CPUExecutionProvider": ("onnxruntime", "ort-nightly"),
         "CUDAExecutionProvider": ("onnxruntime-gpu", "ort-nightly-gpu"),
         "TensorrtExecutionProvider": ("onnxruntime-gpu", "ort-nightly-gpu"),
-        "RocmExecutionProvider": ("onnxruntime-gpu", "ort-nightly-gpu"),
+        "ROCMExecutionProvider": ("onnxruntime-gpu", "ort-nightly-gpu"),
         "OpenVINOExecutionProvider": ("onnxruntime-openvino", None),
         "DmlExecutionProvider": ("onnxruntime-directml", "ort-nightly-directml"),
     }
diff --git a/olive/hardware/accelerator.py b/olive/hardware/accelerator.py
index 06117d2a5..bd6b045cc 100644
--- a/olive/hardware/accelerator.py
+++ b/olive/hardware/accelerator.py
@@ -56,7 +56,7 @@ class AcceleratorLookup:
         "gpu": [
             "DmlExecutionProvider",
             "CUDAExecutionProvider",
-            "ROCmExecutionProvider",
+            "ROCMExecutionProvider",
             "TensorrtExecutionProvider",
             "CPUExecutionProvider",
             "OpenVINOExecutionProvider",