app/controllers/concerns/foreman_wds/discovered_hosts_controller_extensions.rb

+# frozen_string_literal: true
+
+module ForemanWds
+  module DiscoveredHostsControllerExtensions
+    def action_permission
+      return :edit if params[:action] == 'wds_server_selected'
+
+      super
+    end
+  end
+end

app/controllers/concerns/foreman_wds/hosts_controller_extensions.rb

@@ -7,7 +7,7 @@ module ForemanWds
     included do
       before_action :cleanup_wds_params
 
-      define_action_permission %w[wds_server_selected], :edit
+      define_action_permission %w[wds_server_selected], :edit if respond_to? :define_action_permission
     end
 
     def wds_server_selected

lib/foreman_wds/engine.rb

@@ -33,6 +33,7 @@ module ForemanWds
 
       if Foreman::Plugin.installed?('foreman_discovery')
         DiscoveredHostsController.include ForemanWds::HostsControllerExtensions
+        DiscoveredHostsController.prepend ForemanWds::DiscoveredHostsControllerExtensions
       end
     rescue StandardError => e
       Rails.logger.fatal "foreman_wds: skipping engine hook (#{e})"

lib/foreman_wds/version.rb

 # frozen_string_literal: true
 
 module ForemanWds
-  VERSION = '0.0.4'
+  VERSION = '0.0.5'
 end