Minor change to visit.rb