diff --git a/services/git-bridge/src/main/java/uk/ac/ic/wlgitbridge/snapshot/servermock/state/SnapshotAPIStateBuilder.java b/services/git-bridge/src/main/java/uk/ac/ic/wlgitbridge/snapshot/servermock/state/SnapshotAPIStateBuilder.java
index cdc305f2c6..179985f1ea 100644
--- a/services/git-bridge/src/main/java/uk/ac/ic/wlgitbridge/snapshot/servermock/state/SnapshotAPIStateBuilder.java
+++ b/services/git-bridge/src/main/java/uk/ac/ic/wlgitbridge/snapshot/servermock/state/SnapshotAPIStateBuilder.java
@@ -83,30 +83,21 @@ public class SnapshotAPIStateBuilder {
             String projectName,
             JsonObject jsonGetDoc
     ) {
+        String migratedFromId = null;
         if (jsonGetDoc.has("migratedFromId")) {
-            getDoc.put(
-                    projectName,
-                    new GetDocResult(
-                            jsonGetDoc.get("error"),
-                            jsonGetDoc.get("versionID").getAsInt(),
-                            jsonGetDoc.get("createdAt").getAsString(),
-                            jsonGetDoc.get("email").getAsString(),
-                            jsonGetDoc.get("name").getAsString(),
-                            jsonGetDoc.get("migratedFromId").getAsString()
-                    )
-            );
-        } else {
-            getDoc.put(
-                    projectName,
-                    new GetDocResult(
-                            jsonGetDoc.get("error"),
-                            jsonGetDoc.get("versionID").getAsInt(),
-                            jsonGetDoc.get("createdAt").getAsString(),
-                            jsonGetDoc.get("email").getAsString(),
-                            jsonGetDoc.get("name").getAsString()
-                    )
-            );
+            migratedFromId = jsonGetDoc.get("migratedFromId").getAsString();
         }
+        getDoc.put(
+                projectName,
+                new GetDocResult(
+                        jsonGetDoc.get("error"),
+                        jsonGetDoc.get("versionID").getAsInt(),
+                        jsonGetDoc.get("createdAt").getAsString(),
+                        jsonGetDoc.get("email").getAsString(),
+                        jsonGetDoc.get("name").getAsString(),
+                        migratedFromId
+                )
+        );
     }
 
     private void addGetSavedVersForProject(