diff --git a/docs/conf.py b/docs/conf.py
index f65d848..58e2ea0 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -19,9 +19,9 @@
 #
 import os
 import re
+import sys
 import numpydoc
 import sphinx_bootstrap_theme
-import json
 
 import matplotlib as mpl
 mpl.use("Agg")
@@ -29,15 +29,17 @@
 import matplotlib.sphinxext.plot_directive
 import matplotlib.pyplot as plt
 
-import sys
-sys.path.insert(0, os.path.abspath('.'))
-sys.path.insert(0, os.path.abspath('..'))
-sys.path.insert(0, os.path.abspath('../..'))
+# import sys
+# sys.path.insert(0, os.path.abspath('.'))
+# sys.path.insert(0, os.path.abspath('..'))
+# sys.path.insert(0, os.path.abspath('../..'))
 
 import mock
-MOCK_MODULES = ['py2cytoscape.data.cyrest_client.CyRestClient',
-                'rpy2.robjects.packages.importr'
-                'rpy2.robjects.pandas2ri',
+MOCK_MODULES = ['aiohttp',
+                'dask', 'dask.array', 'da',
+                'imageio', 'imageio.v3', 'iio',
+                'xarray', 'xr',
+                #'zarr', 'zarr.storage', 'zarr.storage.BaseStore', # can't mock Zarr for some reason
                 'rpy2.robjects.numpy2ri',
                 'rpy2.robjects.conversion.localconverter']
 
@@ -45,6 +47,8 @@
     sys.modules[mod_name] = mock.Mock()
 
 import pymaid
+import pymaid.stack
+import pymaid.neuron_label
 #from pymaid import cytoscape
 
 from subprocess import check_call as sh
diff --git a/docs/requirements.txt b/docs/requirements.txt
index 56ebd15..b92fe55 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -3,5 +3,5 @@ ipykernel>=5.1.0
 nbsphinx>=0.4.2
 numpydoc>=0.9.1
 sphinx_bootstrap_theme>=0.7.1
-mock
 PyQt5~=5.13.0
+zarr