diff --git a/clients/js/src/types.ts b/clients/js/src/types.ts
index f794508..bc3e07d 100644
--- a/clients/js/src/types.ts
+++ b/clients/js/src/types.ts
@@ -61,10 +61,10 @@ export type GetResponse = {
 
 export type QueryResponse = {
   ids: IDs[];
-  embeddings: null | Embeddings[][];
+  embeddings: null | Embeddings[];
   documents: (null | Document)[][];
   metadatas: (null | Metadata)[][];
-  distances: null | number[][][];
+  distances: null | number[][];
 }
 
 export type AddResponse = {