Fix generator v2