diff --git a/client/alloc_runner.go b/client/alloc_runner.go
index a6dc9de56..92a41ebb0 100644
--- a/client/alloc_runner.go
+++ b/client/alloc_runner.go
@@ -122,7 +122,6 @@ func (r *AllocRunner) RestoreState() error {
 	r.ctx = snap.Context
 	r.allocClientStatus = snap.AllocClientStatus
 	r.allocClientDescription = snap.AllocClientDescription
-	r.taskStates = snap.Alloc.TaskStates
 
 	var snapshotErrors multierror.Error
 	if r.alloc == nil {
@@ -135,6 +134,8 @@ func (r *AllocRunner) RestoreState() error {
 		return e
 	}
 
+	r.taskStates = snap.Alloc.TaskStates
+
 	// Restore the task runners
 	var mErr multierror.Error
 	for name, state := range r.taskStates {