From 659afba17fd3e542a5570401075165e4fc4a40fc Mon Sep 17 00:00:00 2001
From: Yoni Friedman <yoniifriedman@gmail.com>
Date: Thu, 20 Jun 2024 20:04:55 -0400
Subject: [PATCH] #512 fix to avoid dropping some hierarchical addresses from
 discard during update

---
 src/dynamic/update.jl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/dynamic/update.jl b/src/dynamic/update.jl
index 3e3605f5..85ecbcea 100644
--- a/src/dynamic/update.jl
+++ b/src/dynamic/update.jl
@@ -175,10 +175,11 @@ function add_unvisited_to_discard!(discard::DynamicChoiceMap,
                 set_submap!(discard, key, submap)
             else
                 subdiscard = get_submap(discard, key)
+                subdiscard_recursive = isempty(subdiscard) ? choicemap() : subdiscard
                 add_unvisited_to_discard!(
-                    isempty(subdiscard) ? choicemap() : subdiscard,
+                    subdiscard_recursive,
                     subvisited, submap)
-                set_submap!(discard, key, subdiscard)
+                set_submap!(discard, key, subdiscard_recursive)
             end
         end
     end