diff --git a/pyproject.toml b/pyproject.toml
index a7c7c44..3448072 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "multimodalrouter"
-version = "0.1.6"
+version = "0.1.7"
 description = "A graph-based routing library for dynamic routing."
 readme = "README.md"
 license = { file = "LICENSE.md" }
diff --git a/src/multimodalrouter/graph/graph.py b/src/multimodalrouter/graph/graph.py
index bf0f0b4..7788835 100644
--- a/src/multimodalrouter/graph/graph.py
+++ b/src/multimodalrouter/graph/graph.py
@@ -290,7 +290,11 @@ def _generateHubs(self):
                     added.add(row.destination)
 
                 # get extra metrics
-                extra_metrics = {col: getattr(row, col) for col in extra_metric_cols}
+                extra_metrics = {
+                    col: getattr(row, col)
+                    for col in extra_metric_cols
+                    if hasattr(row, col)
+                }
 
                 # link with the extra metrics
                 self._addLink(