diff --git a/scripts/generate_conda_file.py b/scripts/generate_conda_file.py
index 457d541956..2a50195b3d 100644
--- a/scripts/generate_conda_file.py
+++ b/scripts/generate_conda_file.py
@@ -90,10 +90,10 @@
 PIP_PYSPARK = {}
 
 PIP_DARWIN = {
-    "nni": "nni==0.5.2.1",
+    "nni": "nni==0.5.2.1.1",
 }
 PIP_LINUX = {
-    "nni": "nni==0.5.2.1",
+    "nni": "nni==0.5.2.1.1",
 }
 PIP_WIN32 = {}