Fix restore color when unselect