update test for figure_plot