Fixed the handling of x0 in plot_wavefield