Fix generator spec