fixed an example in plot_trace