diff --git a/工具/分类题号字典生成.py b/工具/分类题号字典生成.py
index cf1b4288..cd58e67e 100644
--- a/工具/分类题号字典生成.py
+++ b/工具/分类题号字典生成.py
@@ -1,13 +1,12 @@
 original_id_list = "1:50000"
 
 import re,json
-suffix_words = ["试题"]
+suffix_words = ["试题",r"第[\d]*题"]
 
 def strip_suffix(problem, suf_words):
     origin = problem["origin"]
     for sw in suf_words:
-        if sw in origin:
-            origin = re.sub(sw+r"[\d]*$","",origin)
+        origin = re.sub(sw+r"[\S]*$","",origin)
     return(origin)
 
 def generate_number_set(string):