fix(compiler): Type inference rewriter: Handle return-like operations correctly

fix(compiler): Type inference rewriter: Handle return-like operations correctly #3032