diff --git a/bfps/tools.py b/bfps/tools.py
index 9d39e845fcdb6d28e297a775aa874cba6ff9ae39..ee1e4775f560dab3f2ae46e926d45726ca9c7829 100644
--- a/bfps/tools.py
+++ b/bfps/tools.py
@@ -203,10 +203,11 @@ def padd_with_zeros(
     """
     if (type(odtype) == type(None)):
         odtype = a.dtype
-    assert(a.shape[0] <= n0 and
-           a.shape[1] <= n1 and
+    print(a.shape, n0, n1, n2//2+1)
+    assert(a.shape[0] <= n1 and
+           a.shape[1] <= n0 and
            a.shape[2] <= n2//2+1)
-    b = np.zeros((n0, n1, n2//2 + 1) + a.shape[3:], dtype = odtype)
+    b = np.zeros((n1, n0, n2//2 + 1) + a.shape[3:], dtype = odtype)
     m0 = a.shape[1]
     m1 = a.shape[0]
     m2 = a.shape[2]