diff --git a/Applications/DocXChain/modules/layout_analysis.py b/Applications/DocXChain/modules/layout_analysis.py
index 6702517..5d4a40d 100644
--- a/Applications/DocXChain/modules/layout_analysis.py
+++ b/Applications/DocXChain/modules/layout_analysis.py
@@ -36,9 +36,7 @@ def __init__(self, configs):
 
             # load map information
             map_info = json.load(open(BASE_DIR + '/../../../DocumentUnderstanding/DocXLayout/map_info.json'))
-            category_map = {}
-            for cate, idx in map_info["huntie"]["primary_map"].items():
-                category_map[idx] = cate
+            category_map = {idx: cate for cate, idx in map_info["huntie"]["primary_map"].items()}
 
             self.category_map = category_map
 
@@ -88,4 +86,4 @@ def release(self):
         if self.layout_analyser is not None:
             del self.layout_analyser
 
-        return
\ No newline at end of file
+        return