diff --git a/discretisedfield/line.py b/discretisedfield/line.py index 96803576..2a84c9b1 100644 --- a/discretisedfield/line.py +++ b/discretisedfield/line.py @@ -170,7 +170,7 @@ def point_columns(self): @point_columns.setter def point_columns(self, val): - if len(val) != 3: + if len(val) != len(self._point_columns): msg = f"Cannot change column names with a list of lenght {len(val)}." raise ValueError(msg)