diff --git a/src/View/Pollutants/UndoCommand.py b/src/View/Pollutants/UndoCommand.py index fa612ddc..f27958c8 100644 --- a/src/View/Pollutants/UndoCommand.py +++ b/src/View/Pollutants/UndoCommand.py @@ -34,14 +34,14 @@ class SetNameCommand(QUndoCommand): self._pollutants_lst = pollutants_lst self._index = index - self._old = self._pollutants_lst.get(self._index).name + self._old = self._pollutants_lst[self._index].name self._new = str(new_value) def undo(self): - self._pollutants_lst.get(self._index).name = self._old + self._pollutants_lst[self._index].name = self._old def redo(self): - self._pollutants_lst.get(self._index).name = self._new + self._pollutants_lst[self._index].name = self._new class SetEnabledCommand(QUndoCommand):