diff --git a/Dockerfile b/Dockerfile
index 7bf0554..92681cd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,13 +10,18 @@ RUN apt-get update && \
     gnupg \
     vim \
     curl
+
 RUN wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add -
 RUN echo "deb http://dl.google.com/linux/chrome/deb/ stable main" >> /etc/apt/sources.list.d/google-chrome.list
 RUN apt-get update && \
     apt-get install -y \
     google-chrome-stable
+
 RUN rm -rf /var/lib/apt/lists/*
 
+RUN curl https://sh.rustup.rs -sSf | bash -s -- -y
+ENV PATH="/root/.cargo/bin:${PATH}"
+
 # Create directory structure
 RUN mkdir -p \
     /root/sia \
@@ -49,16 +54,15 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 # SIA core setup
 FROM base AS sia-env
-RUN curl https://sh.rustup.rs -sSf | bash -s -- -y
-ENV PATH="/root/.cargo/bin:${PATH}"
-
-COPY action_schema.xsd /root/sia/action_schema.xsd
-
 COPY ./scripts/setup_binaries.py /root/sia/scripts/
 COPY ./setup.py /root/sia/setup.py
 COPY ./lib /root/sia/lib
 RUN python3 /root/sia/scripts/setup_binaries.py /root/sia/setup.py
 RUN python3 -m venv /root/venvs/sia
+
+COPY action_schema.xsd /root/sia/action_schema.xsd
+RUN mkdir -p /root/sia/lib/xml_schema_validator/src/
+
 RUN --mount=type=cache,target=/root/.cache/pip \
     /root/venvs/sia/bin/pip install -e /root/sia/