diff --git a/banking_breakdown/ui.py b/banking_breakdown/ui.py
index fc68ad7..88ae675 100644
--- a/banking_breakdown/ui.py
+++ b/banking_breakdown/ui.py
@@ -116,14 +116,15 @@ class MainWindow(QMainWindow):
         self._list_widget.itemSelectionChanged.connect(
             self._handle_list_selection_changed)
 
+        self._proxyModel = QSortFilterProxyModel(self)
+        self._table_view.setModel(self._proxyModel)
+
     def set_statement_data(self, df: pd.DataFrame):
         if 'category' not in df.columns:
             df['category'] = [' '] * len(df.index)
 
         model = PandasModel(df)
-        proxyModel = QSortFilterProxyModel(self)
-        proxyModel.setSourceModel(model)
-        self._table_view.setModel(proxyModel)
+        self._proxyModel.setSourceModel(model)
 
         if len(df.columns) < 10:  # Experimentally determined threshold
             # Properly resize columns (takes longer)