diff --git a/src/main.nim b/src/main.nim
index e7b3f41..0235586 100644
--- a/src/main.nim
+++ b/src/main.nim
@@ -110,10 +110,10 @@ for epoch in 0..<Epochs:
             lost = false
         one.params["moves"] = gameOne.moves.float()
         two.params["moves"] = gameTwo.moves.float()
-        networks.sort(cmp=compareNetworks)
-        networks = networks[0..<Take]
-        one = sample(networks)
+    networks.sort(cmp=compareNetworks)
+    networks = networks[0..<Take]
+    one = sample(networks)
+    two = sample(networks)
+    while one == two:
         two = sample(networks)
-        while one == two:
-            two = sample(networks)
-        networks.add(one.crossover(two))
\ No newline at end of file
+    networks.add(one.crossover(two))