Fix wrong return type annotation for function getOutputAST