diff --git a/dax/mds/controller/controller_test.go b/dax/mds/controller/controller_test.go
index 9b204f55b..fd8dea179 100644
--- a/dax/mds/controller/controller_test.go
+++ b/dax/mds/controller/controller_test.go
@@ -9,6 +9,8 @@ import (
 
 	"github.com/featurebasedb/featurebase/v3/dax"
 	"github.com/featurebasedb/featurebase/v3/dax/mds/controller"
+	balancerdb "github.com/featurebasedb/featurebase/v3/dax/mds/controller/balancer/boltdb"
+	schemardb "github.com/featurebasedb/featurebase/v3/dax/mds/schemar/boltdb"
 	daxtest "github.com/featurebasedb/featurebase/v3/dax/test"
 	testbolt "github.com/featurebasedb/featurebase/v3/dax/test/boltdb"
 	"github.com/featurebasedb/featurebase/v3/errors"