diff --git a/setup.py b/setup.py
index d5a119ef38..11dad1d692 100644
--- a/setup.py
+++ b/setup.py
@@ -178,7 +178,7 @@ def run(self):
         },
         install_requires=requirements,
         extras_require={
-            "onnx": ["numpy", "onnx", "onnx-runtime"],
+            "onnx": ["numpy", "onnx", "onnxruntime"],
             "ipex": ["intel_extension_for_pytorch"],
         },
         entry_points={"console_scripts": ["torchserve=ts.model_server:start"]},