Fix class input