diff --git a/pkg/generator/generator.go b/pkg/generator/generator.go
index 7afdf94..957d8eb 100644
--- a/pkg/generator/generator.go
+++ b/pkg/generator/generator.go
@@ -57,14 +57,14 @@ const fileTemplate = `// Code generated by protoc-gen-mcp-go. DO NOT EDIT.
 package {{ .GoPackage }}
 
 import (
+  "connectrpc.com/connect"
   "context"
+  "encoding/json"
   "github.com/mark3labs/mcp-go/mcp"
   mcpserver "github.com/mark3labs/mcp-go/server"
-  "encoding/json"
-  "google.golang.org/protobuf/encoding/protojson"
-  "connectrpc.com/connect"
+  "github.com/northpolesec/protoc-gen-go-mcp/pkg/runtime"
   grpc "google.golang.org/grpc"
-  "github.com/redpanda-data/protoc-gen-go-mcp/pkg/runtime"
+  "google.golang.org/protobuf/encoding/protojson"
 )