Visualize works again

Yay