diff --git a/benchs/bench_fw/descriptors.py b/benchs/bench_fw/descriptors.py
index ab7d09b8f5..5958d6ce93 100644
--- a/benchs/bench_fw/descriptors.py
+++ b/benchs/bench_fw/descriptors.py
@@ -83,6 +83,8 @@ class DatasetDescriptor:
 
     embedding_column: Optional[str] = None
 
+    embedding_id_column: Optional[str] = None
+
     sampling_rate: Optional[float] = None
 
     # sampling column for xdb