diff --git a/test.Dockerfile b/test.Dockerfile
index 37a281c3..a1ed5cd2 100644
--- a/test.Dockerfile
+++ b/test.Dockerfile
@@ -13,8 +13,8 @@ RUN git clone https://github.com/deso-protocol/core.git
 
 WORKDIR /deso/src/core
 RUN git pull && \
-    git checkout mf/pos-merge-20230605 && \
-    git pull origin mf/pos-merge-20230605 # TODO: Revert to `git pull` once core PR is merged.
+    git checkout feature/proof-of-stake && \
+    git pull origin feature/proof-of-stake # TODO: Revert to `git pull` once core PR is merged.
 
 RUN go mod download
 RUN ./scripts/install-relic.sh