diff --git a/gen/generator.go b/gen/generator.go index 1257b44..f5b5b7a 100644 --- a/gen/generator.go +++ b/gen/generator.go @@ -213,7 +213,12 @@ func fixPkgPathVendoring(pkgPath string) string { } func fixAliasName(alias string) string { - alias = strings.Replace(alias, "-", "", -1) + alias = strings.Replace( + strings.Replace(path.Base(pkgPath), ".", "_", -1), + "-", + "_", + -1, + ) return alias }