Fix bugs when change color scheme