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