diff --git a/Dockerfile b/Dockerfile
index 68d47d7..967b81b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -49,7 +49,7 @@ RUN git clone https://github.com/CURENT/dime.git && \
     rm -rf dime \
     && rm -rf /tmp/dime
 
-RUN git clone --single-branch --branch unittesting https://github.com/CURENT/agvis.git && \
+RUN git clone --single-branch --branch master https://github.com/CURENT/agvis.git && \
     cd agvis && \
     python3 -m pip install -e . && \
     cd ..
@@ -66,4 +66,4 @@ COPY . .
 
 # Set the entrypoint and command for the container
 ENTRYPOINT []
-CMD []
\ No newline at end of file
+CMD []