Fixed failing tests in ast_tree_walker spec