diff --git a/models/article.js b/models/article.js
index 41153d7..ca510ad 100644
--- a/models/article.js
+++ b/models/article.js
@@ -49,6 +49,9 @@ let model = db.define('article', {
     {
       fields: ['user_id']
     },
+    {
+      fields: ['problem_id']
+    },
     {
       fields: ['sort_time']
     }
diff --git a/models/judge_state.js b/models/judge_state.js
index 854778e..f1f7933 100644
--- a/models/judge_state.js
+++ b/models/judge_state.js
@@ -76,6 +76,9 @@ let model = db.define('judge_state', {
       },
       {
         fields: ['problem_id'],
+      },
+      {
+        fields: ['task_id'],
       }
     ]
   });