diff --git a/banking_breakdown/ui.py b/banking_breakdown/ui.py
index 9d680a7..71c7c4c 100644
--- a/banking_breakdown/ui.py
+++ b/banking_breakdown/ui.py
@@ -281,6 +281,8 @@ class MainWindow(QMainWindow):
 
         if button == QMessageBox.StandardButton.Yes:
             self._list_widget.takeItem(self._list_widget.row(selected_item))
+            df = self.get_statement_data()
+            df.loc[df['category'] == selected_item.text(), 'category'] = ' '
 
     def _assign_category(self, category: str):
         indexes = self._table_view.selectionModel().selectedRows()