Fixed the visualizer