diff --git a/qubesmanager/tests/test_qube_manager.py b/qubesmanager/tests/test_qube_manager.py index 85ade1b8..c5ea5edf 100644 --- a/qubesmanager/tests/test_qube_manager.py +++ b/qubesmanager/tests/test_qube_manager.py @@ -256,7 +256,9 @@ def test_100_sorting(self): @unittest.mock.patch('qubesmanager.qube_manager.QSettings.setValue') @unittest.mock.patch('qubesmanager.qube_manager.QSettings.sync') def test_101_hide_column(self, mock_sync, mock_settings): - self.dialog.action_is_dvm_template.trigger() + model = self.dialog.qubes_model + action_no = model.columns_indices.index('Is DVM Template') + self.dialog.menu_view.actions()[action_no].trigger() mock_settings.assert_called_with('columns/Is DVM Template', False) self.assertEqual(mock_sync.call_count, 1, "Hidden column not synced")