diff --git a/requirements-ci.txt b/requirements-ci.txt
index 6d11fd9c2..385c8737e 100644
--- a/requirements-ci.txt
+++ b/requirements-ci.txt
@@ -23,5 +23,4 @@ multi_quantization
 
 onnx
 onnxruntime
---extra-index-url https://pypi.ngc.nvidia.com
-onnx_graphsurgeon
+onnx_graphsurgeon -i https://pypi.ngc.nvidia.com
diff --git a/requirements.txt b/requirements.txt
index 2e72d2eb6..9050279de 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,5 +6,4 @@ typeguard
 multi_quantization
 onnx
 onnxruntime
---extra-index-url https://pypi.ngc.nvidia.com
-onnx_graphsurgeon
+onnx_graphsurgeon -i https://pypi.ngc.nvidia.com