diff --git a/benchmarl/__init__.py b/benchmarl/__init__.py
index 5fad7aab..a7bc3127 100644
--- a/benchmarl/__init__.py
+++ b/benchmarl/__init__.py
@@ -6,16 +6,6 @@
 
 import importlib
 
-from benchmarl import (
-    algorithms,
-    benchmark,
-    environments,
-    eval_results,
-    experiment,
-    models,
-    utils,
-)
-
 
 _has_hydra = importlib.util.find_spec("hydra") is not None
 
diff --git a/docs/source/api.rst b/docs/source/api.rst
index 4727592c..55aec41c 100644
--- a/docs/source/api.rst
+++ b/docs/source/api.rst
@@ -4,4 +4,7 @@ API
 .. autosummary::
    :toctree: generated
 
-   benchmarl
+   benchmarl.algorithms
+   benchmarl.models
+   benchmarl.experiment
+   benchmarl.environments