diff --git a/SimPEG/Problem.py b/SimPEG/Problem.py
index dff99a97..b629d723 100644
--- a/SimPEG/Problem.py
+++ b/SimPEG/Problem.py
@@ -160,8 +160,7 @@ class Fields(object):
                 assert hasattr(self, func), 'The alias field function is a string, but it does not exist in the Fields class.'
                 func = getattr(self, func)
             out = func(self._fields[alias][:,ind], srcII)
-
-        if out.shape[1] == 1:
+        if out.shape[0] == out.size:
             out = Utils.mkvc(out)
         return out