diff --git a/watcher/decision_engine/strategy/strategies/workload_balance.py b/watcher/decision_engine/strategy/strategies/workload_balance.py
index bc3b6bffe..f54792f4c 100644
--- a/watcher/decision_engine/strategy/strategies/workload_balance.py
+++ b/watcher/decision_engine/strategy/strategies/workload_balance.py
@@ -205,9 +205,9 @@ class WorkloadBalance(base.WorkloadStabilizationBaseStrategy):
# calculate the available resources
cores_used, mem_used, disk_used = self.calculate_used_resource(
host)
- cores_available = host.vcpus - cores_used
- disk_available = host.disk - disk_used
- mem_available = host.memory - mem_used
+ cores_available = host.vcpu_capacity - cores_used
+ disk_available = host.disk_gb_capacity - disk_used
+ mem_available = host.memory_mb_capacity - mem_used
if (cores_available >= required_cores and
mem_available >= required_mem and
disk_available >= required_disk):
diff --git a/watcher/tests/decision_engine/model/data/scenario_4_with_1_node_no_instance.xml b/watcher/tests/decision_engine/model/data/scenario_4_with_1_node_no_instance.xml
index 722b59250..1132e698c 100644
--- a/watcher/tests/decision_engine/model/data/scenario_4_with_1_node_no_instance.xml
+++ b/watcher/tests/decision_engine/model/data/scenario_4_with_1_node_no_instance.xml
@@ -1,3 +1,3 @@
-
+
diff --git a/watcher/tests/decision_engine/model/data/scenario_6_with_2_nodes.xml b/watcher/tests/decision_engine/model/data/scenario_6_with_2_nodes.xml
index 83189fc32..3d2ac24c6 100644
--- a/watcher/tests/decision_engine/model/data/scenario_6_with_2_nodes.xml
+++ b/watcher/tests/decision_engine/model/data/scenario_6_with_2_nodes.xml
@@ -1,9 +1,9 @@
-
+
-
+