diff --git a/.gitmodules b/.gitmodules
index 896f922c066b214b099ae3fa9eaad6c91f702161..e6601fb0c49b816e4d213f6bc20d6d21b11cbae4 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,24 +1,15 @@
 [submodule "accelerator-wrapper"]
-	path = accelerator-wrapper
-	url = git@git.soton.ac.uk:soclabs/accelerator-wrapper.git
-	[submodule "nanosoc"]
-	path = nanosoc
-	url = git@git.soton.ac.uk:soclabs/nanosoc.git
-	[submodule "socsim"]
-	path = socsim
-	url = git@git.soton.ac.uk:soclabs/socsim.git
-	[submodule "CHIPKIT"]
-	path = CHIPKIT
-	url = git@git.soton.ac.uk:soclabs/CHIPKIT.git
-	[submodule "secworks-sha256"]
-	path = secworks-sha256
-	url = git@git.soton.ac.uk:soclabs/secworks-sha256.git
-	branch = main
-[submodule "secworks-sha-256-system-top/accelerator-wrapper"]
-	branch = main
-[submodule "secworks-sha-256-system-top/CHIPKIT"]
-	branch = main
-[submodule "secworks-sha-256-system-top/nanosoc"]
-	branch = main
-[submodule "secworks-sha-256-system-top/socsim"]
-	branch = main
+path = accelerator-wrapper
+url = git@git.soton.ac.uk:soclabs/accelerator-wrapper.git
+[submodule "nanosoc"]
+path = nanosoc
+url = git@git.soton.ac.uk:soclabs/nanosoc.git
+[submodule "socsim"]
+path = socsim
+url = git@git.soton.ac.uk:soclabs/socsim.git
+[submodule "CHIPKIT"]
+path = CHIPKIT
+url = git@git.soton.ac.uk:soclabs/CHIPKIT.git
+[submodule "secworks-sha256"]
+path = secworks-sha256
+url = git@git.soton.ac.uk:soclabs/secworks-sha256.git
diff --git a/flow/socrepo-init b/flow/socrepo-init
index a14cde6c2168370d9d372a1ca93bd9aae281a6c7..cf24e4c415aabc41d43980953300c67b8d560663 100755
--- a/flow/socrepo-init
+++ b/flow/socrepo-init
@@ -18,9 +18,11 @@ for d in $SOC_TOP_DIR/* ; do
         echo $d
         if [ -f "$d/set_env.sh" ]; then
         # If .git file exists - submodule
-            git config -f .gitmodules submodule.$d.branch main
+            # git config -f .gitmodules submodule.$d.branch main
+            git submodule set-branch --branch main $d
         fi
     fi
 done
 git submodule update --remote --recursive
-git submodule foreach --recursive git checkout main
\ No newline at end of file
+git submodule foreach --recursive git checkout main
+git restore $DESIGN_ROOT/.gitmodules
\ No newline at end of file