From 57e5b7284f3fcb47bae4eea675fc8042baa0493b Mon Sep 17 00:00:00 2001
From: Evan Simkowitz <esimkowitz@users.noreply.github.com>
Date: Thu, 6 Jun 2024 18:07:00 -0700
Subject: [PATCH] Forgot to set generation atom (#28)

---
 frontend/faraday/lib/layoutAtom.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/frontend/faraday/lib/layoutAtom.ts b/frontend/faraday/lib/layoutAtom.ts
index ecd43871f..110edfe6e 100644
--- a/frontend/faraday/lib/layoutAtom.ts
+++ b/frontend/faraday/lib/layoutAtom.ts
@@ -42,7 +42,10 @@ export function withLayoutTreeState<T>(layoutNodeAtom: WritableLayoutNodeAtom<T>
         },
         (get, set, value) => {
             set(pendingActionAtom, value.pendingAction);
-            if (get(generationAtom) !== value.generation) set(layoutNodeAtom, value.rootNode);
+            if (get(generationAtom) !== value.generation) {
+                set(generationAtom, value.generation);
+                set(layoutNodeAtom, value.rootNode);
+            }
         }
     );
 }