diff --git a/docker/Dockerfile.finn b/docker/Dockerfile.finn
index 2ceb1f4195..4f1fc163de 100644
--- a/docker/Dockerfile.finn
+++ b/docker/Dockerfile.finn
@@ -112,6 +112,7 @@ RUN pip install pytest-metadata==1.7.0
 RUN pip install pytest-html==3.0.0
 RUN pip install pytest-html-merger==0.0.8
 RUN pip install pytest-cov==4.1.0
+RUN pip install pyyaml==6.0.1
 
 # extra dependencies from other FINN deps
 # installed in Docker image to make entrypoint script go faster