Fix generator build