diff --git a/ragged/dataset/llama_index.py b/ragged/dataset/llama_index.py
index b809ab6..8ead803 100644
--- a/ragged/dataset/llama_index.py
+++ b/ragged/dataset/llama_index.py
@@ -51,7 +51,7 @@ def get_contexts(self) -> List[TextNode]:
         return self.documents
 
     def get_queries(self) -> List[str]:
-        return self.dataset[self.query_column_name].tolist()
+        return self.to_pandas()[self.query_column_name].tolist()
     
     @property
     def context_column_name(self):