diff --git a/Dockerfile b/Dockerfile
index 87f1f6c..11dc3b3 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,6 @@
 FROM python:3.10-slim-bullseye
-RUN apt-get update -y && apt-get install -y --no-install-recommends bash git openssh graphviz graphviz-dev musl-dev postgresql-dev build-base build-essential gcc python3-dev libffi-dev poppler-utils zlib1g-dev libjpeg-dev libssl-dev && apt-get clean && rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update -y && apt-get install -y --no-install-recommends bash git graphviz graphviz-dev musl-dev build-essential gcc python3-dev libffi-dev poppler-utils zlib1g-dev libjpeg-dev libssl-dev && apt-get clean && rm -rf /var/lib/apt/lists/*
 COPY entrypoint.sh /entrypoint.sh
 
 ENTRYPOINT ["/entrypoint.sh"]
diff --git a/entrypoint.sh b/entrypoint.sh
index bb62965..156bf95 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -27,6 +27,7 @@ _git_changed() {
 
 echo "Installing dependencies..."
 pip3 install -U pip wheel pyparsing pydot
+pip3 install Django django-extensions
 pip3 install -r $INPUT_PIP_PATH
 
 echo "Creating data model..."