diff --git a/src/phoenix/server/api/types/EmbeddingDimension.py b/src/phoenix/server/api/types/EmbeddingDimension.py
index 01c87c26be..b12a0f3788 100644
--- a/src/phoenix/server/api/types/EmbeddingDimension.py
+++ b/src/phoenix/server/api/types/EmbeddingDimension.py
@@ -316,15 +316,11 @@ def UMAPPoints(
             except SchemaError:
                 pass
 
-            try:
-                link_to_data = dataset.get_embedding_link_to_data_column(self.name)[row_id]
-            except SchemaError:
-                pass
+            if col := dataset.get_embedding_link_to_data_column(self.name):
+                link_to_data = col[row_id]
 
-            try:
-                raw_data = dataset.get_embedding_raw_data_column(self.name)[row_id]
-            except SchemaError:
-                pass
+            if col := dataset.get_embedding_raw_data_column(self.name):
+                raw_data = col[row_id]
 
             points[dataset_id].append(
                 UMAPPoint(