diff --git a/integration-test/Dockerfile b/integration-test/Dockerfile
index f4d9e9f..d2c455e 100644
--- a/integration-test/Dockerfile
+++ b/integration-test/Dockerfile
@@ -41,8 +41,8 @@ RUN apt-get update && apt-get install -y \
 RUN apt-get update && apt-get install -y \
   python3.8-dev python3-pip python3.9-dev
 
-RUN python3.8 -m pip install pycapnp psutil pybind11==2.11.2 numpy jinja2 cocotb cocotb_test
-RUN python3.9 -m pip install pycapnp psutil pybind11==2.11.2 numpy jinja2 cocotb cocotb_test
+RUN python3.8 -m pip install pycapnp psutil pybind11==2.11.2 numpy jinja2 cocotb cocotb_test click
+RUN python3.9 -m pip install pycapnp psutil pybind11==2.11.2 numpy jinja2 cocotb cocotb_test click
 
 RUN apt-get update && apt-get install -y tcl