Fix for multi-code-block statements and exit.