diff --git a/docs/environment.yml b/docs/environment.yml
index 2dbfd535..faa2f1ca 100644
--- a/docs/environment.yml
+++ b/docs/environment.yml
@@ -24,3 +24,4 @@ dependencies:
   - attrs>=17.4.0
   - sphinx-copybutton
   - alabaster_jupyterhub
+  - autodoc-traits
diff --git a/docs/requirements.txt b/docs/requirements.txt
index 5f8b447a..44c782f0 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -2,6 +2,7 @@
 # if you change this file
 -r ../requirements.txt
 alabaster_jupyterhub
+autodoc-traits
 recommonmark==0.5.0
 sphinx-copybutton
 sphinx>=1.7
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 6e83f379..45411969 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -17,7 +17,7 @@ extensions = [
     'sphinx.ext.autodoc',
     'sphinx.ext.intersphinx',
     'sphinx.ext.napoleon',
-    'autodoc_traits',
+    'autodoc-traits',
     'sphinx_copybutton',
 ]
 
@@ -37,7 +37,6 @@ from os.path import dirname
 docs = dirname(dirname(__file__))
 root = dirname(docs)
 sys.path.insert(0, root)
-sys.path.insert(0, os.path.join(docs, 'sphinxext'))
 
 import jupyterhub