diff --git a/apps/tasks/tests.py b/apps/tasks/tests.py
index 280d5ba..c3245bc 100644
--- a/apps/tasks/tests.py
+++ b/apps/tasks/tests.py
@@ -529,7 +529,9 @@ def test_handle_uploaded_file(self) -> None:
         handle_uploaded_file(request, self.task.id, self.submission.id)
 
         self.submission.refresh_from_db()
-        self.assertEqual(self.submission.status, "WJ")
+        self.assertEqual(
+            self.submission.status, SubmissionStatus.WAITING_JUDGE
+        )
 
         file_path = os.path.join("apps/tasks/uploads/", "test_file.txt")
         self.assertTrue(os.path.exists(file_path))