diff --git a/python/swiftest/swiftest/simulation_class.py b/python/swiftest/swiftest/simulation_class.py
index 169bc1c0c..9666ca298 100644
--- a/python/swiftest/swiftest/simulation_class.py
+++ b/python/swiftest/swiftest/simulation_class.py
@@ -2409,7 +2409,7 @@ def input_to_array_3d(val,n=None):
                         raise ValueError(f"Argument is an incorrect shape. Expected {(n,3)} or {(3,1)}. Got {val.shape} instead")
                     elif val.shape == (3,):
                         val = np.expand_dims(val,axis=0)
-                elif val.shape != (n,3) or val.shape != (3,n):
+                elif val.shape != (n,3) and val.shape != (3,n):
                     raise ValueError(f"Argument is an incorrect shape. Expected {(n,3)} or {(3,n)}. Got {val.shape} instead")
                 elif val.shape == (3,n):
                     val = val.T