diff --git a/SimPEG/Problem.py b/SimPEG/Problem.py
index dff99a97..81ba0511 100644
--- a/SimPEG/Problem.py
+++ b/SimPEG/Problem.py
@@ -160,9 +160,8 @@ 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:
-            out = Utils.mkvc(out)
+            if not out.size > out.shape[0]:
+                out = Utils.mkvc(out)
         return out
 
     def __contains__(self, other):