diff --git a/lib/utils/vis.py b/lib/utils/vis.py
index db8676e6..85f03fdf 100644
--- a/lib/utils/vis.py
+++ b/lib/utils/vis.py
@@ -239,6 +239,6 @@ def vis_one_image(
                     line, color=colors[len(kp_lines) + 1], linewidth=1.0,
                     alpha=0.7)
 
-        output_name = os.path.basename(im_name) + '.' + ext
-        fig.savefig(os.path.join(output_dir, '{}'.format(output_name)), dpi=dpi)
-        plt.close('all')
+    output_name = os.path.basename(im_name) + '.' + ext
+    fig.savefig(os.path.join(output_dir, '{}'.format(output_name)), dpi=dpi)
+    plt.close('all')