diff --git a/client/devicemanager/manager_test.go b/client/devicemanager/manager_test.go
index 044fb6f26..2cc78da5c 100644
--- a/client/devicemanager/manager_test.go
+++ b/client/devicemanager/manager_test.go
@@ -119,11 +119,12 @@ func baseTestConfig(t *testing.T) (
 	mc := &loader.MockCatalog{}
 
 	// Create the config
+	logger := testlog.HCLogger(t)
 	config = &Config{
-		Logger:        testlog.HCLogger(t),
+		Logger:        logger,
 		PluginConfig:  &base.AgentConfig{},
 		StatsInterval: 100 * time.Millisecond,
-		State:         state.NewMemDB(),
+		State:         state.NewMemDB(logger),
 		Updater:       updateFn,
 		Loader:        mc,
 	}
diff --git a/client/state/db_test.go b/client/state/db_test.go
index 87431d980..230c7ad86 100644
--- a/client/state/db_test.go
+++ b/client/state/db_test.go
@@ -44,7 +44,7 @@ func testDB(t *testing.T, f func(*testing.T, StateDB)) {
 	boltdb, cleanup := setupBoltStateDB(t)
 	defer cleanup()
 
-	memdb := NewMemDB()
+	memdb := NewMemDB(testlog.HCLogger(t))
 
 	impls := []StateDB{boltdb, memdb}