Fix AST/code switcher
Fix AST/code switcher

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