Fixes for visual tree update