Fixed bug in AST creator