diff --git a/src/gui/settings_window.py b/src/gui/settings_window.py
index b80a9f4..459ba8f 100644
--- a/src/gui/settings_window.py
+++ b/src/gui/settings_window.py
@@ -46,11 +46,7 @@ class settingsWindow(QWidget):
         self._num_points.editingFinished.connect(self._expand)
         
         self._param_table = QTableWidget()
-        self._param_table.setColumnCount(2)
-        self._param_table.setRowCount(len(self._data))
-        for i, (key, items) in enumerate(self._data.items()):
-            self._param_table.setItem(i, 0, QTableWidgetItem(key))
-            self._param_table.setItem(i, 1, QTableWidgetItem(str(items[0])))
+        self._restore()
 
         layout = QVBoxLayout()
         header = QLabel(self._name)
@@ -78,8 +74,8 @@ class settingsWindow(QWidget):
     
     def _restore(self) -> None:
         self._param_table.setRowCount(len(self._data))
-        keys = self._data.keys()
-        self._param_table.setColumnCount(len(self._data[keys[0]])+1)
+        key = next(iter(self._data))
+        self._param_table.setColumnCount(len(self._data[key])+1)
         for i, (key, items) in enumerate(self._data.items()):
             self._param_table.setColumnCount(len(self._data[key])+1)
             self._param_table.setItem(i, 0, QTableWidgetItem(key))