diff --git a/api_integration_test.go b/api_integration_test.go
index 8c9f3384..7828d945 100644
--- a/api_integration_test.go
+++ b/api_integration_test.go
@@ -4,6 +4,7 @@ package openai_test
 
 import (
 	"context"
+	"encoding/json"
 	"errors"
 	"io"
 	"os"