AST optimize fix