diff --git a/files/default/neutron-ha-tool.py b/files/default/neutron-ha-tool.py
index b83d7ac0..0f46991d 100755
--- a/files/default/neutron-ha-tool.py
+++ b/files/default/neutron-ha-tool.py
@@ -456,7 +456,7 @@ def agent_alive_id_list(agent_list, agent_type):
     :param agent_list: API response for list_agents()
 
     """
-    return [agent for agent in agent_list
+    return [agent['id'] for agent in agent_list
             if agent['agent_type'] == agent_type and agent['alive'] is True]
 
 
@@ -467,7 +467,7 @@ def agent_dead_id_list(agent_list, agent_type):
     :param agent_list: API response for list_agents()
 
     """
-    return [agent for agent in agent_list
+    return [agent['id'] for agent in agent_list
             if agent['agent_type'] == agent_type and agent['alive'] is False]