Fixing the convergence issue