Handle GeneratorExit correctly