diff --git a/squirrel/sqvm.cpp b/squirrel/sqvm.cpp index 2da6bea..c0c7c98 100644 --- a/squirrel/sqvm.cpp +++ b/squirrel/sqvm.cpp @@ -1006,8 +1006,9 @@ exception_restore: case _OP_YIELD:{ if(ci->_generator) { if(sarg1 != MAX_FUNC_STACKSIZE) temp_reg = STK(arg1); + if (_openouters) CloseOuters(&_stack._vals[_stackbase]); _GUARD(ci->_generator->Yield(this,arg2)); - traps -= ci->_etraps; + traps -= ci->_etraps; if(sarg1 != MAX_FUNC_STACKSIZE) _Swap(STK(arg1),temp_reg);//STK(arg1) = temp_reg; } else { Raise_Error(_SC("trying to yield a '%s',only genenerator can be yielded"), GetTypeName(ci->_generator)); SQ_THROW();}