adjust AST to make it look more like the original AST after optimization