diff --git a/environment.yaml b/environment.yaml
index 91463f0fb1..397abfe9b3 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -12,7 +12,7 @@ dependencies:
   - pip:
       - gradio==3.16.2
       - albumentations==1.3.0
-      - opencv-contrib-python==4.3.0.36
+      - opencv-contrib-python-headless==4.3.0.36
       - imageio==2.9.0
       - imageio-ffmpeg==0.4.2
       - pytorch-lightning==1.5.0