Get rid of GeneratorExit errors