diff --git a/javascript/notification.js b/javascript/notification.js
index 3ee972ae..45ad220b 100644
--- a/javascript/notification.js
+++ b/javascript/notification.js
@@ -26,7 +26,7 @@ onAfterUiUpdate(function() {
     lastHeadImg = headImg;
 
     // play notification sound if available
-    const notificationAudio = gradioApp().querySelector('#audio_notification audio');
+    const notificationAudio = gradioApp().querySelector('#audio_notification #waveform > div')?.shadowRoot?.querySelector('audio');
     if (notificationAudio) {
         notificationAudio.volume = opts.notification_volume / 100.0 || 1.0;
         notificationAudio.play();