Fix: a minor bug in closing lexical scope after parsing return statements.