this seems to fix up the plots.

I only tried it on small simulations though.
100 reps: 3 hrs rather than 30