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)
 	}