Fix issue with the new model passing method