diff --git a/doc/source/changelog.rst b/doc/source/changelog.rst
index ccd0d8c..207603e 100644
--- a/doc/source/changelog.rst
+++ b/doc/source/changelog.rst
@@ -2,6 +2,12 @@
 Changelog
 #########
 
+Version 5.0.1 (2024-06-14)
+~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+- fix: fix return type of Timeline.__iter__ (@domsmrz)
+- improve: remove MatplotlibDeprecationWarning (@kimdwkimdw)
+
 Version 5.0.0 (2022-12-15)
 ~~~~~~~~~~~~~~~~~~~~~~~~~~
 
diff --git a/pyannote/core/notebook.py b/pyannote/core/notebook.py
index 1aa8bec..2138870 100644
--- a/pyannote/core/notebook.py
+++ b/pyannote/core/notebook.py
@@ -126,7 +126,7 @@ def __init__(self):
         self.reset()
 
     def reset(self):
-        from matplotlib.cm import get_cmap
+        from matplotlib.pyplot import get_cmap
 
         linewidth = [3, 1]
         linestyle = ["solid", "dashed", "dotted"]
diff --git a/pyannote/core/timeline.py b/pyannote/core/timeline.py
index 43c1716..1af9151 100755
--- a/pyannote/core/timeline.py
+++ b/pyannote/core/timeline.py
@@ -182,7 +182,7 @@ def __bool__(self):
         """
         return len(self.segments_set_) > 0
 
-    def __iter__(self) -> Iterable[Segment]:
+    def __iter__(self) -> Iterator[Segment]:
         """Iterate over segments (in chronological order)
 
         >>> for segment in timeline: