Fixes tree node highlighting