diff --git a/pkg/db/mysql.go b/pkg/db/mysql.go
index d0d5872..a5b3e1d 100644
--- a/pkg/db/mysql.go
+++ b/pkg/db/mysql.go
@@ -103,6 +103,8 @@ func loadTaskBenchMysqlConn(conn string) *gorm.DB {
 	err = db.AutoMigrate(
 		&model.TaskManagement{},
 		&model.TaskAssignRecords{},
+		&model.TaskBalance{},
+		&model.TaskLog{},
 	)
 
 	if err != nil {