diff --git a/Dockerfile b/Dockerfile
index 7eacb83..9315f48 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -56,6 +56,6 @@ RUN python frontend/fetch-version.py
 USER 1001
 
 # Pre-install npm dependencies for context diagrams
-RUN python -c "from capellambse_context_diagrams import _elkjs; _elkjs._install_required_npm_pkg_versions()"
+RUN python -c "from capellambse_context_diagrams import install_elk; install_elk()"
 
 ENTRYPOINT ["/entrypoint.sh"]