diff --git a/docs/api/layers.md b/docs/api/layers.md
index 620d8cf4..8361035a 100644
--- a/docs/api/layers.md
+++ b/docs/api/layers.md
@@ -1,5 +1,4 @@
 ::: deel.lip.layers
     options:
         show_root_toc_entry: True
-        inherited_members: True
         show_submodules: True
diff --git a/mkdocs.yml b/mkdocs.yml
index 4d05594f..653174e9 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -41,7 +41,13 @@ theme:
 
 plugins:
   - search
-  - mkdocstrings
+  - mkdocstrings:
+      handlers:
+        python:
+          options:
+            merge_init_into_class: true
+            docstring_section_style: spacy
+            separate_signature: true
   - mknotebooks
 
 markdown_extensions:
@@ -62,8 +68,8 @@ markdown_extensions:
       custom_checkbox: true
       clickable_checkbox: true
   - pymdownx.emoji:
-      emoji_index: !!python/name:materialx.emoji.twemoji
-      emoji_generator: !!python/name:materialx.emoji.to_svg
+      emoji_index: !!python/name:material.extensions.emoji.twemoji
+      emoji_generator: !!python/name:material.extensions.emoji.to_svg
 
 extra_css:
   - css/custom.css
diff --git a/setup.py b/setup.py
index 1c1daadd..e480c73c 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,7 @@
     "mkdocstrings-python",
     "mknotebooks",
     "ipython",  # required for Pygments
+    "black",  # required for mkdocs signature rendering
 ]
 
 setuptools.setup(