Fixed bug in AST