diff --git a/Dockerfile b/Dockerfile
index 45567ba..1917c4d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,4 +10,5 @@ RUN dnf install -y git docker sed clang-tools-extra
 
 RUN if [ "$VTK" = "True" ]; then \
         dnf install -y vtk vtk-devel; \ 
+        pip install -U cmakelang pyyaml
     fi