diff --git a/airgun/entities/hostcollection.py b/airgun/entities/hostcollection.py
index 19b61af0f..3de0a724d 100644
--- a/airgun/entities/hostcollection.py
+++ b/airgun/entities/hostcollection.py
@@ -242,9 +242,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()