small fix for plotting script