diff --git a/mongodb_proxy.py b/mongodb_proxy.py
index b784300..dfe2bdf 100644
--- a/mongodb_proxy.py
+++ b/mongodb_proxy.py
@@ -17,6 +17,7 @@
 from functools import wraps
 import logging
 import pymongo
+import pymongo.collection
 import time
 
 log = logging.getLogger(__name__)