From d04fc2647cd14c35d99a797a1afc569c00517d30 Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Tue, 2 Feb 2016 15:18:35 -0800
Subject: [PATCH] Fix test

---
 client/task_runner_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/client/task_runner_test.go b/client/task_runner_test.go
index 52c8c73b3..f0f6c92e2 100644
--- a/client/task_runner_test.go
+++ b/client/task_runner_test.go
@@ -49,7 +49,7 @@ func testTaskRunner(restarts bool) (*MockTaskStateUpdater, *TaskRunner) {
 
 	ctx := driver.NewExecContext(allocDir, alloc.ID)
 	rp := structs.NewRestartPolicy(structs.JobTypeService)
-	restartTracker := newRestartTracker(rp)
+	restartTracker := newRestartTracker(rp, alloc.Job.Type)
 	if !restarts {
 		restartTracker = noRestartsTracker()
 	}