diff --git a/environments/huggingface.yml b/environments/huggingface.yml
index 1bef463d..e2f5ccc5 100644
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@@ -5,6 +5,7 @@ channels:
   - defaults
 dependencies:
   - colorama
+  - flask=2.2.3
   - flask-socketio=5.3.2
   - flask-session=0.4.0
   - python-socketio=5.7.2
diff --git a/environments/rocm.yml b/environments/rocm.yml
index 47d29e07..a0e370ca 100644
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@@ -4,6 +4,7 @@ channels:
   - defaults
 dependencies:
   - colorama
+  - flask=2.2.3
   - flask-socketio=5.3.2
   - flask-session=0.4.0
   - python-socketio=5.7.2