1 files changed, 1 insertions, 2 deletions

diff --git a/music_sampler/music_file.py b/music_sampler/music_file.py
index fa6293d..4ba65e3 100644
--- a/music_sampler/music_file.py
+++ b/music_sampler/music_file.py
@@ -241,8 +241,7 @@ class MusicFile:
                 self.current_audio_segment = new_audio_segment
             self.stop_playing()
             if wait:
-                if set_wait_id is not None:
-                    self.mapping.add_wait_id(set_wait_id, self.wait_event)
+                self.mapping.add_wait(self.wait_event, wait_id=set_wait_id)
                 self.wait_end()
         else:
             self.stopped()