Fixed a bug that causes blow-up at large step size

Thus, the radnom step size part is restored, since it is not the cause
of the blow-up and can speed up the computation.