Fix class error