fixed the color staying bug