diff --git a/src/fireedge/src/client/models/Host.js b/src/fireedge/src/client/models/Host.js
index 76212715af..7f2d8a192e 100644
--- a/src/fireedge/src/client/models/Host.js
+++ b/src/fireedge/src/client/models/Host.js
@@ -97,7 +97,8 @@ export const getAllocatedInfo = (host) => {
  * @returns {Array} List of hugepages sizes from resource
  */
 export const getHugepageSizes = (host) => {
-  const numaNodes = [host?.HOST_SHARE?.NUMA_NODES?.NODE ?? []].flat()
+  const wrapHost = Array.isArray(host) ? host : [host]
+  const numaNodes = [wrapHost?.HOST_SHARE?.NUMA_NODES?.NODE ?? []].flat()
 
   return numaNodes
     .filter((node) => node?.NODE_ID && node?.HUGEPAGE)
@@ -137,10 +138,11 @@ export const getKvmCpuModels = (hosts = []) => {
  * @returns {Array} List of KVM Machines from the pool
  */
 export const getKvmCpuFeatures = (hosts = []) => {
-  const machineTypes = hosts
-    .filter((host) => host?.TEMPLATE?.HYPERVISOR === HYPERVISORS.kvm)
-    .map((host) => host.TEMPLATE?.KVM_CPU_FEATURES.split(','))
-    .flat()
+  const wrapHosts = Array.isArray(hosts) ? hosts : [hosts]
+  const machineTypes = wrapHosts
+    ?.filter((host) => host?.TEMPLATE?.HYPERVISOR === HYPERVISORS?.kvm)
+    ?.map((host) => host?.TEMPLATE?.KVM_CPU_FEATURES?.split(',') ?? [])
+    ?.flat()
 
   // Removes the repeated
   return [...new Set(machineTypes)]