small fix in plot