diff --git a/nomad/mock/mock.go b/nomad/mock/mock.go
index b3a468d64..3c2bb4f5e 100644
--- a/nomad/mock/mock.go
+++ b/nomad/mock/mock.go
@@ -235,6 +235,11 @@ func Alloc() *structs.Allocation {
 				},
 			},
 		},
+		TaskStates: map[string]*structs.TaskState{
+			"web": &structs.TaskState{
+				State: structs.TaskStatePending,
+			},
+		},
 		Job:           Job(),
 		DesiredStatus: structs.AllocDesiredStatusRun,
 		ClientStatus:  structs.AllocClientStatusPending,