Fixed the broken visualization.