diff --git a/main_test.go b/main_test.go
index f55d2eb..192325f 100644
--- a/main_test.go
+++ b/main_test.go
@@ -15,6 +15,6 @@ func TestMain(t *testing.T) {
 	// out, _ := json.Marshal(pkg)
 	// tools.WriteFile("test.txt", []string{string(out)}, true)
 
-	list := model.Generate(pkg)
+	list, _ := model.Generate(pkg)
 	fmt.Println(list)
 }