Fix model build