Fix model check on collection save