From 8842bf6087718bd4e1c1610ae00c3c358b999318 Mon Sep 17 00:00:00 2001
From: Simon Gardling <titaniumtown@proton.me>
Date: Tue, 18 Feb 2025 12:18:41 -0500
Subject: [PATCH] improve update_root_coord

---
 src/complexagent.rs | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/complexagent.rs b/src/complexagent.rs
index ccc6c57..a2d6f16 100644
--- a/src/complexagent.rs
+++ b/src/complexagent.rs
@@ -262,9 +262,14 @@ impl FutureMoves {
         self.arena
             .iter()
             .enumerate()
-            .find_map(|(idx, node)| {
-                (node.parent == self.current_root && node.i == i && node.j == j).then_some(idx)
+            .filter(|(_, node)| {
+                node.parent == self.current_root
+                    && self.current_root.is_some()
+                    && node.i == i
+                    && node.j == j
             })
+            .next()
+            .map(|x| x.0)
             .inspect(|&root| self.update_root_idx(root))
             .is_some()
     }