diff --git a/tests/sdk_test.go b/tests/sdk_test.go
index 5f685b6..2fc54a3 100644
--- a/tests/sdk_test.go
+++ b/tests/sdk_test.go
@@ -400,11 +400,11 @@ func TestNewClient(t *testing.T) {
 			},
 		}
 
-		toolResults := make([]*cohere.ChatRequestToolResultsItem, 0)
+		toolResults := make([]*cohere.ToolResult, 0)
 
 		for _, toolCall := range toolsResponse.ToolCalls {
 			result := localTools[toolCall.Name](toolCall.Parameters["day"].(string))
-			toolResult := &cohere.ChatRequestToolResultsItem{
+			toolResult := &cohere.ToolResult{
 				Call:    toolCall,
 				Outputs: *result,
 			}