diff --git a/pkg/config/schema.go b/cmd/baton-databricks/config/schema.go
similarity index 100%
rename from pkg/config/schema.go
rename to cmd/baton-databricks/config/schema.go
diff --git a/pkg/config/schema_test.go b/cmd/baton-databricks/config/schema_test.go
similarity index 100%
rename from pkg/config/schema_test.go
rename to cmd/baton-databricks/config/schema_test.go
diff --git a/cmd/baton-databricks/main.go b/cmd/baton-databricks/main.go
index a89b6d85..6a738967 100644
--- a/cmd/baton-databricks/main.go
+++ b/cmd/baton-databricks/main.go
@@ -5,7 +5,7 @@ import (
 	"fmt"
 	"os"
 
-	"github.com/conductorone/baton-databricks/pkg/config"
+	"github.com/conductorone/baton-databricks/cmd/baton-databricks/config"
 	configSchema "github.com/conductorone/baton-sdk/pkg/config"
 	"github.com/conductorone/baton-sdk/pkg/connectorbuilder"
 	"github.com/conductorone/baton-sdk/pkg/types"
diff --git a/pkg/databricks/client.go b/pkg/databricks/client.go
index 070a4c9a..4b6bc51b 100644
--- a/pkg/databricks/client.go
+++ b/pkg/databricks/client.go
@@ -7,7 +7,7 @@ import (
 	"net/url"
 	"strings"
 
-	"github.com/conductorone/baton-databricks/pkg/config"
+	"github.com/conductorone/baton-databricks/cmd/baton-databricks/config"
 	v2 "github.com/conductorone/baton-sdk/pb/c1/connector/v2"
 	"github.com/conductorone/baton-sdk/pkg/uhttp"
 	"github.com/spf13/viper"