fix broken generator test