ENH: fix for visualizations
ENH: fix for visualizations