Fix generator is not exported