diff --git a/submission/tasks.py b/submission/tasks.py
index 25476e36..0e958e7b 100644
--- a/submission/tasks.py
+++ b/submission/tasks.py
@@ -1,9 +1,9 @@
 # coding=utf-8
-from huey.djhuey import task
+from huey.djhuey import db_task
 
 from judge_dispatcher.tasks import JudgeDispatcher
 
 
-@task()
+@db_task()
 def _judge(submission, time_limit, memory_limit, test_case_id, is_waiting_task=False):
     JudgeDispatcher(submission, time_limit, memory_limit, test_case_id).judge(is_waiting_task)
\ No newline at end of file