Fix color change bug when notification is clicked or screen rotates