diff --git a/apiserver/internal/models/task.go b/apiserver/internal/models/task.go
index 73f5d562..319f2e42 100644
--- a/apiserver/internal/models/task.go
+++ b/apiserver/internal/models/task.go
@@ -30,8 +30,8 @@ type TaskHistory struct {
 }
 
 type TaskLabel struct {
-	TaskID  int `json:"task_id"`
-	LabelID int `json:"label_id"`
+	TaskID  int `json:"task_id" gorm:"primaryKey"`
+	LabelID int `json:"label_id" gorm:"primaryKey"`
 }
 
 type CreateTaskReq struct {
diff --git a/apiserver/internal/utils/migration/migration.go b/apiserver/internal/utils/migration/migration.go
index f620e40f..a24c0aee 100644
--- a/apiserver/internal/utils/migration/migration.go
+++ b/apiserver/internal/utils/migration/migration.go
@@ -12,6 +12,7 @@ func Migration(db *gorm.DB) error {
 		models.AppToken{},
 		models.Label{},
 		models.Task{},
+		models.TaskLabel{},
 		models.TaskHistory{},
 		models.NotificationSettings{},
 		models.Notification{},