diff --git a/installer/Dockerfile b/installer/Dockerfile
index b2499b1b8..953a345b7 100644
--- a/installer/Dockerfile
+++ b/installer/Dockerfile
@@ -35,10 +35,10 @@ ARG DOCKER_IMAGE_TAG=dev \
     GITHUB_COMMIT
 
 ENV MAXKB_VERSION ${DOCKER_IMAGE_TAG} (build at ${BUILD_AT}, commit: ${GITHUB_COMMIT})
-
+WORKDIR /opt/maxkb/app
+COPY --from=stage-build /opt/maxkb /opt/maxkb
 COPY --from=stage-build /opt/py3 /opt/py3
 COPY --from=web-build ui /opt/maxkb/app/ui
-COPY --from=stage-build /opt/maxkb /opt/maxkb
 COPY --from=vector-model /opt/maxkb/app/model /opt/maxkb/app/model
 
 ENV LANG=en_US.UTF-8 \
diff --git a/installer/Dockerfile-python-pg b/installer/Dockerfile-python-pg
index 71ae8b9d7..eb2501468 100644
--- a/installer/Dockerfile-python-pg
+++ b/installer/Dockerfile-python-pg
@@ -2,6 +2,7 @@ FROM postgres:15.6-bookworm
 
 ARG DEPENDENCIES="                    \
         curl                          \
+        vim                           \
         python3.11-mini               \
         python3.11-venv               \
         postgresql-15-pgvector"