fixed the visualization graph