diff --git a/amundsen_common/models/table.py b/amundsen_common/models/table.py
index 592c728174..1493413774 100644
--- a/amundsen_common/models/table.py
+++ b/amundsen_common/models/table.py
@@ -152,7 +152,7 @@ class Table:
     schema: str
     name: str
     tags: List[Tag] = []
-    badges: Optional[List[Tag]] = []
+    badges: List[Badge] = []
     table_readers: List[Reader] = []
     description: Optional[str] = None
     columns: List[Column]
diff --git a/setup.py b/setup.py
index acb1d39dd4..6402f36355 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name='amundsen-common',
-    version='0.4.0',
+    version='0.5.0',
     description='Common code library for Amundsen',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',