diff --git a/app/models/concerns/foreman_wds/host_extensions.rb b/app/models/concerns/foreman_wds/host_extensions.rb
index b0a5220a091b8585b17009e685796fa9e834d38f..6249c5c256e61c03e451f495d431d057cf6cd82c 100644
--- a/app/models/concerns/foreman_wds/host_extensions.rb
+++ b/app/models/concerns/foreman_wds/host_extensions.rb
@@ -41,11 +41,13 @@ module ForemanWds
 
     def capabilities
       return super + [:wds] if compute_resource && (os.nil? || os.family == 'Windows')
+
       super
     end
 
     def bare_metal_capabilities
       return super + [:wds] if os.nil? || os.family == 'Windows'
+
       super
     end
 
@@ -57,6 +59,12 @@ module ForemanWds
       provision_method == 'wds'
     end
 
+    def pxe_build?
+      return true if wds_build?
+
+      super
+    end
+
     def wds?
       managed? && wds_build? && SETTINGS[:unattended]
     end
diff --git a/app/models/concerns/foreman_wds/nic_extensions.rb b/app/models/concerns/foreman_wds/nic_extensions.rb
index 040402bb8944d987568f79c6dc0504d20ec8e45b..5487ddfbedead376ca2172a1c006e201d85caff5 100644
--- a/app/models/concerns/foreman_wds/nic_extensions.rb
+++ b/app/models/concerns/foreman_wds/nic_extensions.rb
@@ -10,7 +10,9 @@ module ForemanWds
     def boot_server
       return super if host.nil? || !host.wds? || host.wds_facet.nil?
 
-      host.wds_server.next_server_ip if host.build? # TODO: Support choosing local boot method
+      return host.wds_server.next_server_ip if host.build? # TODO: Support choosing local boot method
+
+      super
     end
 
     def dhcp_records