Fix visitor bug