diff --git a/src/aiida/__init__.py b/src/aiida/__init__.py
index 00f51cff95..bb62ae9c65 100644
--- a/src/aiida/__init__.py
+++ b/src/aiida/__init__.py
@@ -106,7 +106,10 @@ def get_file_header(comment_char: str = '# ') -> str:
 
 
 def load_ipython_extension(ipython):
-    """Load the AiiDA IPython extension, using ``%load_ext aiida``."""
+    """Load the AiiDA IPython extension, using ``%load_ext aiida``.
+
+    :param ipython: InteractiveShell instance. If omitted, the global InteractiveShell is used.
+    """
     from aiida.tools.ipython.ipython_magics import load_ipython_extension
 
     load_ipython_extension(ipython)
diff --git a/src/aiida/tools/ipython/ipython_magics.py b/src/aiida/tools/ipython/ipython_magics.py
index 486ee50481..342053eb55 100644
--- a/src/aiida/tools/ipython/ipython_magics.py
+++ b/src/aiida/tools/ipython/ipython_magics.py
@@ -205,7 +205,6 @@ def register_ipython_extension(ipython=None):
     The %aiida IPython extension provides the same environment as the `verdi shell`.
 
     :param ipython: InteractiveShell instance. If omitted, the global InteractiveShell is used.
-
     """
     if ipython is None:
         ipython = get_ipython()