diff --git a/Ohmpi.py b/Ohmpi.py
index bc38c981175ccc8c1deb5902279601cb5880b349..6278553d1e0cf2eb4a22edf0f4e5627c06243e55 100644
--- a/Ohmpi.py
+++ b/Ohmpi.py
@@ -194,6 +194,9 @@ Main loop
 """
 N=read_quad("ABMN.txt",pardict.get("nb_electrodes")) # load quadripole file
 
+if N.ndim == 1:
+    N = N.reshape(1, 4)
+
 for g in range(0,pardict.get("nbr_meas")): # for time-lapse monitoring
 
     for i in range(0,N.shape[0]): # loop over quadripoles