Fix emission of `while` in return position
Fix emission of `while` in return position

cucapra/braid,guoyiteng/braid,guoyiteng/braid,guoyiteng/braid,cucapra/braid,cucapra/braid,cucapra/braid,guoyiteng/braid,guoyiteng/braid,cucapra/braid,guoyiteng/braid,cucapra/braid