Fixes the bug where stepping out of a function would show the wrong