diff --git a/main.py b/main.py
index c894f961bb8bba570b52192f54bec15e3f19cf7a..1eb7374004380fc3ef549111e4e08fc39c639b2d 100644
--- a/main.py
+++ b/main.py
@@ -1,3 +1,4 @@
+import os
 from src.top import Top, InputMeta, OutputMeta
 
 
@@ -48,5 +49,6 @@ def test_build_csdn_trend_top10_tag_race():
 
 
 if __name__ == "__main__":
+    os.makedirs('pub', exist_ok=True)
     # test_build_csdn_ask_top10_tag_race()
     test_build_csdn_trend_top10_tag_race()