diff --git a/client/driver/driver_test.go b/client/driver/driver_test.go
index 8cd44b331ae..c64ee1e7943 100644
--- a/client/driver/driver_test.go
+++ b/client/driver/driver_test.go
@@ -88,6 +88,7 @@ type testContext struct {
 // It is up to the caller to call AllocDir.Destroy to cleanup.
 func testDriverContexts(t *testing.T, task *structs.Task) *testContext {
 	cfg := testConfig()
+	cfg.Node = mock.Node()
 	allocDir := allocdir.NewAllocDir(testLogger(), filepath.Join(cfg.AllocDir, structs.GenerateUUID()))
 	if err := allocDir.Build(); err != nil {
 		t.Fatalf("AllocDir.Build() failed: %v", err)