diff --git a/client_generator.go b/client_generator.go
index b4dc0cb..b970710 100644
--- a/client_generator.go
+++ b/client_generator.go
@@ -1,3 +1,3 @@
 package main
 
-//go:generate go run github.com/oapi-codegen/oapi-codegen/v2/cmd/oapi-codegen -package=ollama_client -generate=types,client,spec -o=ollama_client/api.gen.go openapi.yaml
+//go:generate go run github.com/oapi-codegen/oapi-codegen/v2/cmd/oapi-codegen -package=ollama_client -generate=types,client,spec -o=./internal/ollama_client/api.gen.go openapi.yaml
diff --git a/cmd/main.go b/cmd/main.go
index 9285545..e6d6a57 100644
--- a/cmd/main.go
+++ b/cmd/main.go
@@ -37,7 +37,6 @@ import (
 
 	ollamav1 "github.com/StartUpNationLabs/simple-ollama-operator/api/v1"
 	"github.com/StartUpNationLabs/simple-ollama-operator/internal/controller"
-	_ "github.com/StartUpNationLabs/simple-ollama-operator/ollama_client"
 	// +kubebuilder:scaffold:imports
 )
 
diff --git a/internal/controller/model_controller.go b/internal/controller/model_controller.go
index bc32fdd..b8e3d49 100644
--- a/internal/controller/model_controller.go
+++ b/internal/controller/model_controller.go
@@ -18,8 +18,7 @@ package controller
 
 import (
 	"context"
-	"github.com/StartUpNationLabs/simple-ollama-operator/ollama_client"
-
+	"github.com/StartUpNationLabs/simple-ollama-operator/internal/ollama_client"
 	"k8s.io/apimachinery/pkg/runtime"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
diff --git a/ollama_client/api.gen.go b/internal/ollama_client/api.gen.go
similarity index 100%
rename from ollama_client/api.gen.go
rename to internal/ollama_client/api.gen.go