fixed issue with disappearing squares in tree_map