fixed a problem with the figures in the GUI