diff --git a/docker/Dockerfile b/docker/Dockerfile
index a18913c..6d9ee91 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -12,6 +12,7 @@ RUN apt-get update && apt-get upgrade -y \
 
 COPY environment.yml /tmp/
 RUN conda update -y -n base conda \
+    && echo '    - pyvirtualdisplay' >> /tmp/environment.yml \
     && conda env create -f /tmp/environment.yml \
     && conda clean -y -t \
     && rm /tmp/environment.yml
diff --git a/environment.yml b/environment.yml
index e0256fe..24b17f9 100644
--- a/environment.yml
+++ b/environment.yml
@@ -36,7 +36,6 @@ dependencies:
     - opencv-python==4.1.1.26
     - psutil==5.6.3
     - pyglet==1.3.2
-    - pyvirtualdisplay==0.2.4
     - tensorboard==2.0.0
     - tensorflow==2.0.0
     - tensorflow-addons==0.6.0