AM gets removed from knownAppMasters map when the task count becomes 0. Task count becomes 0 during fragment execution as unregisterTask happens when the task is actually scheduled to run. This will make umbilical connection to AM to be closed. Any new rejections or task kills will have to recreate the connection. This can be improved by remove the AM info from knownAppMaster after query completion.