diff --git a/cmd/protoc-gen-gogottrpc/main.go b/cmd/protoc-gen-gogottrpc/main.go
index 7016202..23e1fcb 100644
--- a/cmd/protoc-gen-gogottrpc/main.go
+++ b/cmd/protoc-gen-gogottrpc/main.go
@@ -4,7 +4,7 @@ import (
 	"github.com/gogo/protobuf/protoc-gen-gogo/descriptor"
 	"github.com/gogo/protobuf/vanity"
 	"github.com/gogo/protobuf/vanity/command"
-	_ "github.com/stevvooe/ttrpc"
+	_ "github.com/stevvooe/ttrpc/plugin"
 )
 
 func main() {
diff --git a/generator.go b/plugin/generator.go
similarity index 99%
rename from generator.go
rename to plugin/generator.go
index b27f47a..e8495b1 100644
--- a/generator.go
+++ b/plugin/generator.go
@@ -1,4 +1,4 @@
-package ttrpc
+package plugin
 
 import (
 	"strings"