Fix class comment