diff --git a/index.d.ts b/index.d.ts
index 783c07a..f110a5a 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -7,9 +7,9 @@ declare module 'apollo-datasource-mongodb' {
     Model as MongooseModel
   } from 'mongoose'
 
-  export type Collection<T> = T extends Document
+  export type Collection<T, U = MongoCollection<T>> = T extends Document
     ? MongooseCollection
-    : MongoCollection<T>
+    : U
 
   export type Model<T> = T extends Document ? MongooseModel<T> : undefined