Fix: some labels in figure_draw.py