diff --git a/Dockerfile b/Dockerfile
index 7457dcf..6d482b5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -34,6 +34,7 @@ RUN apt-get update -y && apt-get install -y \
 RUN bash ./scripts/setup_omnizart.sh
 RUN pip install --no-cache-dir -r requirements.txt
 RUN pip install -U numpy
+RUN pip install click==8.0.4
 RUN python test.py
 
 # Run the web service on container startup. Here we use the gunicorn