diff --git a/src/mongopersist/datamanager.py b/src/mongopersist/datamanager.py
index 6957f5d..3b5c41a 100644
--- a/src/mongopersist/datamanager.py
+++ b/src/mongopersist/datamanager.py
@@ -278,8 +278,8 @@ def dump(self, obj):
             del self._registered_objects[id(obj)]
         return res
 
-    def load(self, dbref):
-        return self._reader.get_ghost(dbref)
+    def load(self, dbref, klass=None):
+        return self._reader.get_ghost(dbref, klass)
 
     def reset(self):
         root = self.root