diff --git a/python/test/binclass/eptest_binclass.py b/python/test/binclass/eptest_binclass.py
index 4a2c7db..ad1098d 100644
--- a/python/test/binclass/eptest_binclass.py
+++ b/python/test/binclass/eptest_binclass.py
@@ -173,6 +173,7 @@
     opts = abt.helpers.Struct()
     opts.imode = imode
     opts.maxit = 20       # Max. number sweeps
+    #opts.maxit = 40       # Max. number sweeps
     opts.deltaeps = 1e-4  # Convergence threshold
     opts.damp = 0.        # No damping
     opts.caveps = 1e-5
diff --git a/src/eptools/wrap/eptwrap_epupdate_parallel.cc b/src/eptools/wrap/eptwrap_epupdate_parallel.cc
index 5a620a0..3eb9fe5 100644
--- a/src/eptools/wrap/eptwrap_epupdate_parallel.cc
+++ b/src/eptools/wrap/eptwrap_epupdate_parallel.cc
@@ -63,7 +63,8 @@ void eptwrap_epupdate_parallel(int ain,int aout,W_IARRAY(potids),
     //cout << "Wrap: Done createPotentialManager" << endl; // DEBUG!
     totsz=ncmu;
     W_CHKSIZE(crho,totsz,"CRHO");
-    if (potMan->numArgumentGroup(EPScalarPotential::atypeUnivariate)!=totsz)
+    if (potMan->numArgumentGroup(EPScalarPotential::atypeUnivariate)!=
+	potMan->size())
       W_RETERROR(1,"All potentials must be in group 'atypeUnivariate'");
     if (ain>7) {
       if (updind==0)