small visitor code structure change