diff --git a/server/src/com/cloud/agent/manager/AgentManagerImpl.java b/server/src/com/cloud/agent/manager/AgentManagerImpl.java
index a7ed857e63f..77d9854415f 100755
--- a/server/src/com/cloud/agent/manager/AgentManagerImpl.java
+++ b/server/src/com/cloud/agent/manager/AgentManagerImpl.java
@@ -921,7 +921,7 @@ public class AgentManagerImpl implements AgentManager, HandlerFactory, Manager {
         }
     }
 
-    public void removeAgent(AgentAttache attache, Status nextState, Event event, Boolean investigate, boolean ha) {
+    public void removeAgent(AgentAttache attache, Status nextState) {
         if (attache == null) {
             return;
         }
@@ -946,12 +946,7 @@ public class AgentManagerImpl implements AgentManager, HandlerFactory, Manager {
             removed.disconnect(nextState);
         }
         
-        HostVO host = _hostDao.findById(hostId);
-        if (ha && event != null && investigate != null) {
-        	if (!event.equals(Event.PrepareUnmanaged) && !event.equals(Event.HypervisorVersionChanged) && (host.getStatus() == Status.Alert || host.getStatus() == Status.Down)) {
-                _haMgr.scheduleRestartForVmsOnHost(host, investigate);
-            }
-        }
+
         
         for (Pair<Integer, Listener> monitor : _hostMonitors) {
             if (s_logger.isDebugEnabled()) {
@@ -1012,7 +1007,7 @@ public class AgentManagerImpl implements AgentManager, HandlerFactory, Manager {
         HostVO host = _hostDao.findById(hostId);
         if (host == null) {
             s_logger.warn("Can't find host with " + hostId);
-            removeAgent(attache, Status.Removed, event, investigate, ha);
+            removeAgent(attache, Status.Removed);
             return true;
 
         }
@@ -1022,7 +1017,7 @@ public class AgentManagerImpl implements AgentManager, HandlerFactory, Manager {
                 s_logger.debug("Host " + hostId + " is already " + currentState);
             }
             if (currentState != Status.PrepareForMaintenance) {
-                removeAgent(attache, currentState, event, investigate, ha);
+                removeAgent(attache, currentState);
             }
             return true;
         }
@@ -1110,8 +1105,11 @@ public class AgentManagerImpl implements AgentManager, HandlerFactory, Manager {
         if (s_logger.isDebugEnabled()) {
             s_logger.debug("Deregistering link for " + hostId + " with state " + nextState);
         }
-        removeAgent(attache, nextState, event, investigate, ha);
+        removeAgent(attache, nextState);
         _hostDao.disconnect(host, event, _nodeId);
+        if (ha && event != null && !event.equals(Event.PrepareUnmanaged) && !event.equals(Event.HypervisorVersionChanged) && (host.getStatus() == Status.Alert || host.getStatus() == Status.Down)) {
+            _haMgr.scheduleRestartForVmsOnHost(host, investigate);
+        }
 
         return true;
     }
diff --git a/server/src/com/cloud/agent/manager/ClusteredAgentManagerImpl.java b/server/src/com/cloud/agent/manager/ClusteredAgentManagerImpl.java
index 9f30a141505..63bed4d621b 100755
--- a/server/src/com/cloud/agent/manager/ClusteredAgentManagerImpl.java
+++ b/server/src/com/cloud/agent/manager/ClusteredAgentManagerImpl.java
@@ -169,7 +169,7 @@ public class ClusteredAgentManagerImpl extends AgentManagerImpl implements Clust
                             if (s_logger.isInfoEnabled()) {
                                 s_logger.info(host + " is detected down, but we have a forward attache running, disconnect this one before launching the host");
                             }
-                            removeAgent(agentattache, Status.Disconnected, null, null, false);
+                            removeAgent(agentattache, Status.Disconnected);
                         } else {
                             continue;
                         }
@@ -709,12 +709,12 @@ public class ClusteredAgentManagerImpl extends AgentManagerImpl implements Clust
     }
 
     @Override
-    public void removeAgent(AgentAttache attache, Status nextState, Event event, Boolean investigate, boolean ha) {
+    public void removeAgent(AgentAttache attache, Status nextState) {
         if (attache == null) {
             return;
         }
 
-        super.removeAgent(attache, nextState, event, investigate, ha);
+        super.removeAgent(attache, nextState);
     }
 
     @Override