Fixed color save/restore bug