AST visitor bugfixes