Fix model check