diff --git a/scripts/Engineering/gui/engineering_diffraction/tabs/fitting/data_handling/test/test_data_presenter.py b/scripts/Engineering/gui/engineering_diffraction/tabs/fitting/data_handling/test/test_data_presenter.py
index 09311b4b79fbb7ba772aab511426b9f50777441f..671dbf4dfae6137cc1ae2d97d58c81a2b79d3147 100644
--- a/scripts/Engineering/gui/engineering_diffraction/tabs/fitting/data_handling/test/test_data_presenter.py
+++ b/scripts/Engineering/gui/engineering_diffraction/tabs/fitting/data_handling/test/test_data_presenter.py
@@ -73,6 +73,7 @@ class FittingDataPresenterTest(unittest.TestCase):
         model_dict = {"ENGINX_1_bank_1": "ws1", "ENGINX_2_bank_South": "ws2"}
         self.model.get_loaded_workspaces.return_value = model_dict
         self.model.get_sample_log_from_ws.return_value = "bankOrRunNumber"
+        self.presenter.plot_added_notifier = mock.MagicMock()
 
         self.presenter._on_worker_success("info")
 
@@ -85,6 +86,7 @@ class FittingDataPresenterTest(unittest.TestCase):
         model_dict = {"invalid": "ws1", "invalid2": "ws2"}
         self.model.get_loaded_workspaces.return_value = model_dict
         self.model.get_sample_log_from_ws.side_effect = RuntimeError("No sample logs present")
+        self.presenter.plot_added_notifier = mock.MagicMock()
 
         self.presenter._on_worker_success("info")
 
@@ -98,6 +100,7 @@ class FittingDataPresenterTest(unittest.TestCase):
         model_dict = {"INSTRUMENT_10_bank_2": "ws1", "INSTRUMENT_20_bank_1": "ws2"}
         self.model.get_loaded_workspaces.return_value = model_dict
         self.model.get_sample_log_from_ws.side_effect = RuntimeError("No sample logs present")
+        self.presenter.plot_added_notifier = mock.MagicMock()
 
         self.presenter._on_worker_success("info")
 
@@ -133,12 +136,14 @@ class FittingDataPresenterTest(unittest.TestCase):
         model_dict = {"name1": "ws1", "name2": "ws2"}
         self.model.get_loaded_workspaces.return_value = model_dict
         self.presenter.row_numbers = {"name1": 0, "name2": 1}
+        self.presenter.all_plots_removed_notifier = mock.MagicMock()
 
         self.presenter.rename_workspace("name1", "new")
 
         self.assertEqual({"new": "ws1", "name2": "ws2"}, model_dict)
         self.assertTrue("new" in self.presenter.row_numbers)
         self.assertFalse("name1" is self.presenter.row_numbers)
+        self.assertEqual(1, self.presenter.all_plots_removed_notifier.notify_subscribers.call_count)
 
     def test_rename_workspace_not_tracked(self):
         model_dict = {"name1": "ws1", "name2": "ws2"}
@@ -192,6 +197,7 @@ class FittingDataPresenterTest(unittest.TestCase):
         self.model.get_loaded_workspaces.return_value = model_dict
         self.view.remove_selected.return_value = [0, 2]
         self.presenter.plot_removed_notifier = mock.MagicMock()
+        self.presenter.plot_added_notifier = mock.MagicMock()
 
         self.presenter._remove_selected_tracked_workspaces()
 
@@ -201,6 +207,7 @@ class FittingDataPresenterTest(unittest.TestCase):
         self.assertEqual(self.presenter.row_numbers, test_dict)
         self.assertEqual(model_dict, {"name2": "ws2"})
         self.assertEqual(2, self.presenter.plot_removed_notifier.notify_subscribers.call_count)
+        self.assertEqual(1, self.presenter.plot_added_notifier.notify_subscribers.call_count)
 
     def test_handle_table_cell_changed_checkbox_ticked(self):
         mocked_table_item = mock.MagicMock()