Fix model init