Fix codegen for dispatch calls with yielding blocks