Fix class colour bug.