More AST improvements for QLS