diff --git a/lib/foreman_wds/engine.rb b/lib/foreman_wds/engine.rb
index 0dd95334eb86928c1ee2cc1d4042407a9872ee7f..4bbedc7520de7ad1bd6fab21d34a7499f85c9b8e 100644
--- a/lib/foreman_wds/engine.rb
+++ b/lib/foreman_wds/engine.rb
@@ -83,9 +83,13 @@ module ForemanWds
       HostsController.include ForemanWds::HostsControllerExtensions
       UnattendedController.prepend ForemanWds::UnattendedControllerExtensions
 
-      ComputeResource.providers.each do |_k, const|
+      ComputeResource.providers.each_value do |const|
         Kernel.const_get(const).send(:prepend, ForemanWds::ComputeResourceExtensions)
       end
+
+      if Foreman::Plugin.installed?('foreman_discovery')
+        DiscoveredHostsController.include ForemanWds::HostsControllerExtensions
+      end
     rescue StandardError => e
       Rails.logger.fatal "foreman_wds: skipping engine hook (#{e})"
     end