diff --git a/km3io/tools.py b/km3io/tools.py
index 4f2b4d170674da384e585d76cc4173de63a0b380..5c95062a23c844070736f0ef2e5b0c14d0043475 100644
--- a/km3io/tools.py
+++ b/km3io/tools.py
@@ -233,7 +233,7 @@ def get_multiplicity(tracks, rec_stages):
 
     Returns
     -------
-    class km3io.offline.OfflineBranch
+    km3io.offline.OfflineBranch
         tracks branch with the desired reconstruction stages only.
     """
     return tracks[mask(tracks, rec_stages)]