diff --git a/gen/generator.go b/gen/generator.go
index f5b5b7a..4a89f90 100644
--- a/gen/generator.go
+++ b/gen/generator.go
@@ -214,7 +214,7 @@ func fixPkgPathVendoring(pkgPath string) string {
 
 func fixAliasName(alias string) string {
 	alias = strings.Replace(
-		strings.Replace(path.Base(pkgPath), ".", "_", -1),
+		strings.Replace(alias, ".", "_", -1),
 		"-",
 		"_",
 		-1,