diff --git a/internal/util/flowgraph/flow_graph_test.go b/internal/util/flowgraph/flow_graph_test.go
index 63cd91fa2c..88c84af8a3 100644
--- a/internal/util/flowgraph/flow_graph_test.go
+++ b/internal/util/flowgraph/flow_graph_test.go
@@ -5,7 +5,6 @@ import (
 	"fmt"
 	"log"
 	"math"
-	"math/big"
 	"math/rand"
 	"sync"
 	"testing"
@@ -139,9 +138,15 @@ func sendMsgFromCmd(ctx context.Context, fg *TimeTickedFlowGraph) {
 			}
 			// assert result
 			expect := math.Pow(num, 2) + math.Sqrt(num)
-			if big.NewFloat(res) != big.NewFloat(expect) {
-				fmt.Println(res)
-				fmt.Println(math.Pow(num, 2) + math.Sqrt(num))
+			resBits := math.Float64bits(res)
+			expBits := math.Float64bits(expect)
+			var diffBits uint64
+			if resBits >= expBits {
+				diffBits = resBits - expBits
+			} else {
+				diffBits = expBits - resBits
+			}
+			if diffBits > 2 {
 				panic("wrong answer")
 			}
 		}
diff --git a/scripts/run_go_unittest.sh b/scripts/run_go_unittest.sh
index b77186c72c..877c807083 100755
--- a/scripts/run_go_unittest.sh
+++ b/scripts/run_go_unittest.sh
@@ -13,5 +13,5 @@ SCRIPTS_DIR="$( cd -P "$( dirname "$SOURCE" )" && pwd )"
 # ignore Minio,S3 unittes
 MILVUS_DIR="${SCRIPTS_DIR}/../internal/"
 echo $MILVUS_DIR
-go test -cover "${MILVUS_DIR}/kv/..." "${MILVUS_DIR}/msgstream/..." "${MILVUS_DIR}/master/..." "${MILVUS_DIR}/querynode/..." "${MILVUS_DIR}/storage" "${MILVUS_DIR}/proxy/..." "${MILVUS_DIR}/writenode/..." -failfast
+go test -cover "${MILVUS_DIR}/kv/..." "${MILVUS_DIR}/msgstream/..." "${MILVUS_DIR}/master/..." "${MILVUS_DIR}/querynode/..." "${MILVUS_DIR}/storage" "${MILVUS_DIR}/proxy/..." "${MILVUS_DIR}/writenode/..." "${MILVUS_DIR}/util/..." -failfast
 #go test -cover "${MILVUS_DIR}/kv/..." "${MILVUS_DIR}/msgstream/..." "${MILVUS_DIR}/master/..." "${MILVUS_DIR}/querynode/..." -failfast