diff --git a/common/src/main/java/org/opensearch/ml/common/connector/functions/postprocess/BedrockRerankPostProcessFunction.java b/common/src/main/java/org/opensearch/ml/common/connector/functions/postprocess/BedrockRerankPostProcessFunction.java
index 2b65943b71..46468b1c80 100644
--- a/common/src/main/java/org/opensearch/ml/common/connector/functions/postprocess/BedrockRerankPostProcessFunction.java
+++ b/common/src/main/java/org/opensearch/ml/common/connector/functions/postprocess/BedrockRerankPostProcessFunction.java
@@ -52,7 +52,7 @@ public List<ModelTensor> process(List<Map<String, Object>> rerankResults) {
                             .name("similarity")
                             .shape(new long[] { 1 })
                             .data(new Number[] { scores[i].doubleValue() })
-                            .dataType(MLResultDataType.FLOAT32)
+                            .dataType(MLResultDataType.FLOAT64)
                             .build()
                     );
             }