Fix class color