diff --git a/.gitmodules b/.gitmodules
index 3cd33570..3f14c362 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,9 +1,10 @@
 [submodule "flownet2"]
 	path = flownet2
 	url = https://github.com/NVIDIA/flownet2-pytorch.git
-[submodule "codes/models/flownet2"]
-	path = codes/models/flownet2
-	url = https://github.com/NVIDIA/flownet2-pytorch.git
 [submodule "codes/switched_conv"]
 	path = codes/switched_conv
 	url = https://github.com/neonbjb/SwitchedConvolutions.git
+[submodule "codes/models/flownet2"]
+	path = codes/models/flownet2
+	url = https://github.com/neonbjb/flownet2-pytorch.git
+	branch = master