diff --git a/src/mongo_driver_3/collection.clj b/src/mongo_driver_3/collection.clj
index f9cbb30..e9c1518 100644
--- a/src/mongo_driver_3/collection.clj
+++ b/src/mongo_driver_3/collection.clj
@@ -5,7 +5,8 @@
   (:import (com.mongodb MongoNamespace)
            (com.mongodb.client MongoDatabase MongoCollection ClientSession)
            (com.mongodb.client.model IndexModel)
-           (java.util List)))
+           (java.util List)
+           (org.bson Document)))
 
 (set! *warn-on-reflection* true)