Fixing model build script