Fixed visualization bug