Merge pull request #938 from lulucas/master

modelgen hook docs fixed
99designs · Nov 25, 2019 · e747d92 · e747d92
2 parents 15b3058 + 63be1d5
commit e747d92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/content/recipes/modelgen-hook.md b/docs/content/recipes/modelgen-hook.md
@@ -25,7 +25,7 @@ import (
 )
 
 // Defining mutation function
-func mutateHook(b *ModelBuild) *ModelBuild {
+func mutateHook(b *modelgen.ModelBuild) *modelgen.ModelBuild {
 	for _, model := range b.Models {
 		for _, field := range model.Fields {
 			field.Tag += ` orm_binding:"` + model.Name + `.` +  field.Name + `"`
@@ -49,7 +49,7 @@ func main() {
 
 	err = api.Generate(cfg,
 		api.NoPlugins(),
-		api.AddPlugin(p),
+		api.AddPlugin(&p),
 	)
 	if err != nil {
 		fmt.Fprintln(os.Stderr, err.Error())