diff --git a/docs/source/conf.py b/docs/source/conf.py
index 5af959c7..98937fdd 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -13,6 +13,8 @@
 sys.path.insert(0, os.path.abspath('../../PyNite'))
 sys.path.insert(0, os.path.abspath('../../PyNite/FEModel3D'))
 
+from PyNite import FEModel3D
+
 # autodoc settings
 autodoc_mock_imports = ['numpy']  # Mock the numpy import