diff --git a/environments/huggingface.yml b/environments/huggingface.yml
index 138993dd..669d5d6d 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 c3e95496..dc2dd40e 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