Fix model generator example