diff --git a/docs/conf.py b/docs/conf.py
index 75b21bf..030a11d 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -26,7 +26,7 @@ def __getattr__(cls, name):
         return MagicMock()
 
 MOCK_MODULES = ['numpy', 'pandas', 'pyspark', 'pyspark.sql', 'pyspark.conf',
-                'fastparquet']
+                'pyspark.sql.functions', 'pyspark.sql.types','fastparquet']
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)