Working on AST: some tests fail