small fix on visualize_images