diff --git a/airgun/entities/hostcollection.py b/airgun/entities/hostcollection.py
index 1c12be815..4ea72ad22 100644
--- a/airgun/entities/hostcollection.py
+++ b/airgun/entities/hostcollection.py
@@ -240,9 +240,7 @@ def change_assigned_content(self, entity_name, lce, content_view):
         view.lce.fill({lce: True})
         view.content_view.fill(content_view)
         view.assign.click()
-        self.browser.handle_alert()
-        view.flash.assert_no_error()
-        view.flash.dismiss()
+        view.dialog.confirm_dialog.click()
         task_view = HostCollectionActionTaskDetailsView(view.browser)
         task_view.progressbar.wait_for_result()
         return task_view.read()