Fix class definition