From b097191ec6e7585cc77a815e431ca44a3e937bbf Mon Sep 17 00:00:00 2001
From: Alex Dadgar <alex.dadgar@gmail.com>
Date: Mon, 26 Sep 2016 15:35:59 -0700
Subject: [PATCH] Test fix

---
 client/driver/driver_test.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/client/driver/driver_test.go b/client/driver/driver_test.go
index b8b1c2889..d62d4afb4 100644
--- a/client/driver/driver_test.go
+++ b/client/driver/driver_test.go
@@ -84,7 +84,7 @@ func testDriverContexts(task *structs.Task) (*DriverContext, *ExecContext) {
 	alloc := mock.Alloc()
 	execCtx := NewExecContext(allocDir, alloc.ID)
 
-	taskEnv, err := GetTaskEnv(allocDir, cfg.Node, task, alloc)
+	taskEnv, err := GetTaskEnv(allocDir, cfg.Node, task, alloc, "")
 	if err != nil {
 		return nil, nil
 	}
@@ -119,7 +119,7 @@ func TestDriver_GetTaskEnv(t *testing.T) {
 
 	alloc := mock.Alloc()
 	alloc.Name = "Bar"
-	env, err := GetTaskEnv(nil, nil, task, alloc)
+	env, err := GetTaskEnv(nil, nil, task, alloc, "")
 	if err != nil {
 		t.Fatalf("GetTaskEnv() failed: %v", err)
 	}