Fix demo to get a convergence