diff --git a/generator.go b/generator.go
index b90bbc79..f20548c1 100644
--- a/generator.go
+++ b/generator.go
@@ -94,8 +94,8 @@ type Logger interface {
 // Generator code generator
 type Generator struct {
 	Config
-	Data   map[string]*genInfo                  //gen query data
-	models map[string]*generate.QueryStructMeta //gen model data
+	Data   map[string]*genInfo                  // gen query data
+	models map[string]*generate.QueryStructMeta // gen model data
 
 	logger Logger
 }
@@ -315,12 +315,14 @@ func (g *Generator) generateQueryFile() (err error) {
 			err := g.generateSingleQueryFile(info)
 			if err != nil {
 				errChan <- err
+				return
 			}
 
 			if g.WithUnitTest {
 				err = g.generateQueryUnitTestFile(info)
 				if err != nil { // do not panic
 					g.db.Logger.Error(context.Background(), "generate unit test fail: %s", err)
+					return
 				}
 			}
 		}(info)