Trying to fix issues with visualization.