Fix prob on get_plot_size();