fixed imports
export augmentAst