Still having trouble tracing the non-stochasticity bug