Fix class syntax