fixed issues with visualize