clean up an incorrect generator usage