diff --git a/common/src/main/java/dev/langchain4j/store/embedding/TextSegmentConvert.java b/common/src/main/java/dev/langchain4j/store/embedding/TextSegmentConvert.java
index 4d8f496d6..adee0c857 100644
--- a/common/src/main/java/dev/langchain4j/store/embedding/TextSegmentConvert.java
+++ b/common/src/main/java/dev/langchain4j/store/embedding/TextSegmentConvert.java
@@ -42,6 +42,6 @@ public class TextSegmentConvert {
         if (Objects.isNull(textSegment) || Objects.isNull(textSegment.metadata())) {
             return null;
         }
-        return textSegment.metadata().get(QUERY_ID);
+        return textSegment.metadata().getString(QUERY_ID);
     }
 }