diff --git a/test.Dockerfile b/test.Dockerfile
index 89bc12bb..3dbaf5f6 100644
--- a/test.Dockerfile
+++ b/test.Dockerfile
@@ -10,17 +10,12 @@ ENV PATH="/usr/local/go/bin:${PATH}"
 # Declare an ARG for the branch name with a default value of "main"
 ARG BRANCH_NAME=main
 
-# Declare an ARG for the branch name with a default value of "main"
-ARG BRANCH_NAME=main
-
 WORKDIR /deso/src
 
 RUN git clone https://github.com/deso-protocol/core.git
 
 WORKDIR /deso/src/core
-RUN git pull && \
-    git checkout feature/proof-of-stake && \
-    git pull origin feature/proof-of-stake # TODO: Revert to `git pull` once core PR is merged.
+RUN git pull
 
 RUN go mod download