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