diff --git a/pyproject.toml b/pyproject.toml
index 99751e7..2a008e9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,7 +9,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "swankit"
-version = "0.1.8"
+version = "0.2.0"
 dynamic = ["readme", "dependencies"]
 description = "Base toolkit for SwanLab"
 license = "Apache-2.0"
diff --git a/swankit/callback/models/key.py b/swankit/callback/models/key.py
index 3133c02..348eef8 100644
--- a/swankit/callback/models/key.py
+++ b/swankit/callback/models/key.py
@@ -14,7 +14,7 @@
 import os
 
 ColumnClass = Literal["CUSTOM", "SYSTEM"]
-SectionType = Literal["PINNED", "HIDDEN", "PUBLIC", "SYSTEM"]
+SectionType = Literal["PINNED", "HIDDEN", "PUBLIC", "SYSTEM", "CUSTOM"]
 YRange = Optional[Tuple[Optional[float], Optional[float]]]