From 3f7f95de8c44ae20426e3d27d9d22f5a170c642e Mon Sep 17 00:00:00 2001
From: rowanc1 <rowan@visiblegeology.com>
Date: Tue, 15 Apr 2014 09:19:49 -0700
Subject: [PATCH] magDipole Bug

---
 simpegEM/Utils/Sources/magneticDipole.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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