[REF] models with new API