Fix generator foo