Move visualize to separate test