Minor tweak for visualize.