Fixed the visualizations