Fix generator specs