fix visitor bug