Fixes to the visualizations