diff --git a/routes/classify.py b/routes/classify.py
index 09d38a6..c81a243 100644
--- a/routes/classify.py
+++ b/routes/classify.py
@@ -5,7 +5,7 @@ import uuid
 import zipfile
 import rarfile
 import shutil
-from flask import Blueprint, request, jsonify, current_app, send_file, g, session
+from flask import Blueprint, request, jsonify, current_app, send_file, g, session, after_this_request
 from werkzeug.utils import secure_filename
 import mysql.connector
 from utils.model_service import text_classifier # Assuming text_classifier is initialized correctly elsewhere
@@ -634,7 +634,7 @@ def download_multiple_documents():
 
         logger.info(f"用户 {user_id} 请求下载 {files_added} 个文档打包为 {zip_filename}")
         # Use after_this_request to delete the temp file after sending
-        @current_app.after_request
+        @after_this_request
         def remove_file(response):
             try:
                 if zip_path and os.path.exists(zip_path):