diff --git a/llama-proxy.go b/llama-proxy.go
index a073e28..b948dd6 100644
--- a/llama-proxy.go
+++ b/llama-proxy.go
@@ -6,7 +6,6 @@ import (
 	"net/http"
 	"os"
 
-	"github.com/mostlygeek/go-llama-cpp-proxy/config"
 	"github.com/mostlygeek/go-llama-cpp-proxy/proxy"
 )
 
@@ -17,7 +16,7 @@ func main() {
 
 	flag.Parse() // Parse the command-line flags
 
-	config, err := config.LoadConfig(*configPath)
+	config, err := proxy.LoadConfig(*configPath)
 	if err != nil {
 		fmt.Printf("Error loading config: %v\n", err)
 		os.Exit(1)
diff --git a/config/config.go b/proxy/config.go
similarity index 96%
rename from config/config.go
rename to proxy/config.go
index 93dfaa1..d6bf3b4 100644
--- a/config/config.go
+++ b/proxy/config.go
@@ -1,4 +1,4 @@
-package config
+package proxy
 
 import (
 	"os"
diff --git a/proxy/proxy.go b/proxy/proxy.go
index 84738a4..899e2b5 100644
--- a/proxy/proxy.go
+++ b/proxy/proxy.go
@@ -13,20 +13,18 @@ import (
 	"sync"
 	"syscall"
 	"time"
-
-	"github.com/mostlygeek/go-llama-cpp-proxy/config"
 )
 
 type ProxyManager struct {
 	sync.Mutex
 
-	config       *config.Config
+	config       *Config
 	currentCmd   *exec.Cmd
 	currentModel string
 	currentProxy string
 }
 
-func New(config *config.Config) *ProxyManager {
+func New(config *Config) *ProxyManager {
 	return &ProxyManager{config: config}
 }