diff --git a/dblp_crawler/graph.py b/dblp_crawler/graph.py
index 8ce1b4b..faebebe 100644
--- a/dblp_crawler/graph.py
+++ b/dblp_crawler/graph.py
@@ -12,7 +12,7 @@ class Graph(metaclass=abc.ABCMeta):
     def __init__(self, pid_list: list[str], journal_list: list[str]) -> None:
         self.persons: dict[str, Optional[DBLPPerson]] = {pid: None for pid in pid_list}
         self.summarized_person: set[str] = set()
-        self.publications: dict[str, Publication] = set()
+        self.publications: set[str] = set()
         self.init_journals = journal_list
         self.journals_inited = False
         self.total_author_succ_count, self.total_author_fail_count = 0, 0
diff --git a/setup.py b/setup.py
index 1df726c..0685e19 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
 
 setup(
     name='dblp_crawler',
-    version='2.0',
+    version='2.0.1',
     author='yindaheng98',
     author_email='yindaheng98@gmail.com',
     url='https://github.com/yindaheng98/dblp-crawler',