fixing model passing