diff --git a/integration-test/go.mod b/integration-test/go.mod
index 6beb9b5..e897e09 100644
--- a/integration-test/go.mod
+++ b/integration-test/go.mod
@@ -3,7 +3,7 @@ module integration-test
 go 1.22.0
 
 require (
-	github.com/azaurus1/go-pinot-api v0.1.4
+	github.com/azaurus1/go-pinot-api v0.6.1
 	github.com/google/uuid v1.6.0
 	github.com/hamba/avro/v2 v2.20.1
 	github.com/stretchr/testify v1.9.0
diff --git a/integration-test/go.sum b/integration-test/go.sum
index 5a13c4a..8aa2838 100644
--- a/integration-test/go.sum
+++ b/integration-test/go.sum
@@ -10,8 +10,8 @@ github.com/Microsoft/go-winio v0.6.1 h1:9/kr64B9VUZrLm5YYwbGtUJnMgqWVOdUAXu6Migc
 github.com/Microsoft/go-winio v0.6.1/go.mod h1:LRdKpFKfdobln8UmuiYcKPot9D2v6svN5+sAH+4kjUM=
 github.com/Microsoft/hcsshim v0.11.4 h1:68vKo2VN8DE9AdN4tnkWnmdhqdbpUFM8OF3Airm7fz8=
 github.com/Microsoft/hcsshim v0.11.4/go.mod h1:smjE4dvqPX9Zldna+t5FG3rnoHhaB7QYxPRqGcpAD9w=
-github.com/azaurus1/go-pinot-api v0.1.4 h1:ofZiioIVycYojmG0o5+4EWyze7WxuOETtOWr8oBhyFU=
-github.com/azaurus1/go-pinot-api v0.1.4/go.mod h1:uSL8T9cQCVP5U0MKBKCYuntMpIqjvZRgGk/epgSAQgw=
+github.com/azaurus1/go-pinot-api v0.6.1 h1:4wDwDgM7qyUn/faF5+slCsna8MpQ5yNTktF42VEXjT8=
+github.com/azaurus1/go-pinot-api v0.6.1/go.mod h1:Zxi9gp5nAlU95XH1FQlAXLzH/cLGdX6KaisGtVF56t0=
 github.com/cenkalti/backoff/v4 v4.2.1 h1:y4OZtCnogmCPw98Zjyt5a6+QwPLGkiQsYW5oUqylYbM=
 github.com/cenkalti/backoff/v4 v4.2.1/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
 github.com/containerd/containerd v1.7.13 h1:wPYKIeGMN8vaggSKuV1X0wZulpMz4CrgEsZdaCyB6Is=