Fixed bug in visit method
Visitor test