diff --git a/docs/conf.py b/docs/conf.py
index 7dc07fb..0abf3ff 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -17,7 +17,11 @@
 import os
 import sys
 import sys
-from unittest.mock import MagicMock
+
+try:
+    from mock import Mock as MagicMock
+except:
+    from unittest.mock import MagicMock
 
 class Mock(MagicMock):
     @classmethod