diff --git a/field.py b/field.py index cce2b91ed3c543b90499de61cf75524528ae63e7..c24bdc1c49b4b4e931795d770cc8f5b3e5f8de20 100644 --- a/field.py +++ b/field.py @@ -347,8 +347,8 @@ class Field: def center_vector(self): index_shape = self.index_shape if len(index_shape) == 0: - return self.center - elif len(index_shape) == 1: + return sp.Matrix([self.center]) + if len(index_shape) == 1: return sp.Matrix([self(i) for i in range(index_shape[0])]) elif len(index_shape) == 2: def cb(*args):