diff --git a/dockerfile b/dockerfile
index 1ddf0e7..51e1b78 100644
--- a/dockerfile
+++ b/dockerfile
@@ -25,6 +25,7 @@ RUN pip3 install --upgrade --no-cache-dir onnxruntime-gpu
 
 # Install skelda
 RUN pip3 install --upgrade --no-cache-dir scipy
+RUN apt-get update && apt-get install -y --no-install-recommends xvfb
 COPY ./skelda/ /skelda/
 RUN pip3 install --no-cache-dir -e /skelda/