diff --git a/simpegEM/Utils/Sources/magneticDipole.py b/simpegEM/Utils/Sources/magneticDipole.py
index db492a94..94381aaf 100644
--- a/simpegEM/Utils/Sources/magneticDipole.py
+++ b/simpegEM/Utils/Sources/magneticDipole.py
@@ -37,4 +37,6 @@ def MagneticDipoleVectorPotential(txLoc, obsLoc, component, dipoleMoment=(0., 0.
         mCr = np.cross(m, dR)
         r = np.sqrt((dR**2).sum(axis=1))
         A[:, i] = -(mu_0/(4*pi)) * mCr[:,dimInd]/(r**3)
-    return A
\ No newline at end of file
+    if nTx == 1:
+        return A.flatten()
+    return A