diff --git a/statapp/models/input_values_model.py b/statapp/models/input_values_model.py
index 6f5629c..0d770ea 100644
--- a/statapp/models/input_values_model.py
+++ b/statapp/models/input_values_model.py
@@ -10,7 +10,7 @@ class InputValuesModel(EditableTableModel):
         super().__init__(data)
 
     def getHorizontalHeader(self):
-        return yx_header(self.rowCount(QModelIndex()))
+        return yx_header(self.columnCount(QModelIndex()))
 
     def getY(self):
         return self._data[:, 0]