How to remove a node during map call back

I am trying to use the map function to 'remove' a node on a match condition.. but sort of as expected it is modifying the map as it is being iterated and getting item not found.

Other than generating a list of all the paths and then deleting them after the map is done.. is there a special way to remove a node from the treeMap during a map ?

I tried to mark this as a question but not sure how.