From 950c80cce698f1caecfab2e01884cfea7606de5a Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Thu, 17 Dec 2015 15:33:57 -0800
Subject: [PATCH] Copy the task states from the client update

---
 nomad/state/state_store.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nomad/state/state_store.go b/nomad/state/state_store.go
index 30ee87259821..ff41dc9e4ffa 100644
--- a/nomad/state/state_store.go
+++ b/nomad/state/state_store.go
@@ -556,6 +556,7 @@ func (s *StateStore) UpdateAllocFromClient(index uint64, alloc *structs.Allocati
 	// Pull in anything the client is the authority on
 	copyAlloc.ClientStatus = alloc.ClientStatus
 	copyAlloc.ClientDescription = alloc.ClientDescription
+	copyAlloc.TaskStates = alloc.TaskStates
 
 	// Update the modify index
 	copyAlloc.ModifyIndex = index