diff --git a/test/test_nlte.py b/test/test_nlte.py
index 6fc00282..1eea383f 100644
--- a/test/test_nlte.py
+++ b/test/test_nlte.py
@@ -117,7 +117,7 @@ def test_dll(lfs_atmo, lfs_nlte):
     for lr, li in zip(linerefs, lineindices):
         if lr[0] != -1 and lr[1] != -1:
             counter += 1
-            libsme.InputDepartureCoefficients(bmat[:, lr].T, li)
+            libsme.InputDepartureCoefficients(bmat[:, lr], li)
 
     flags = libsme.GetNLTEflags()
     assert np.any(flags)
@@ -138,13 +138,13 @@ def test_dll(lfs_atmo, lfs_nlte):
         libsme.InputDepartureCoefficients(None, 0)
 
     with pytest.raises(TypeError):
-        libsme.InputDepartureCoefficients(bmat[:, [0, 1]].T, 0.1)
+        libsme.InputDepartureCoefficients(bmat[:, [0, 1]], 0.1)
 
     with pytest.raises(ValueError):
         libsme.InputDepartureCoefficients([0, 1], 10)
 
     with pytest.raises(RuntimeError):
-        libsme.InputDepartureCoefficients(bmat[:, [0, 1]].T, -10)
+        libsme.InputDepartureCoefficients(bmat[:, [0, 1]], -10)
 
 
 @pytest.fixture