fixed bug in .visualize()