diff --git a/enjoy/src/renderer/components/messages/assistant-message.tsx b/enjoy/src/renderer/components/messages/assistant-message.tsx
index 9091354d..fea5d287 100644
--- a/enjoy/src/renderer/components/messages/assistant-message.tsx
+++ b/enjoy/src/renderer/components/messages/assistant-message.tsx
@@ -169,7 +169,7 @@ export const AssistantMessageComponent = (props: {
                   new URL(props.href ?? "");
                   props.target = "_blank";
                   props.rel = "noopener noreferrer";
-                } catch (e) {}
+                } catch (e) { }
 
                 return <a {...props}>{children}</a>;
               },
@@ -252,7 +252,7 @@ export const AssistantMessageComponent = (props: {
               <DownloadIcon
                 data-tooltip-id="global-tooltip"
                 data-tooltip-content={t("download")}
-                data-testid="message-start-shadow"
+                data-testid="message-download"
                 onClick={handleDownload}
                 className="w-3 h-3 cursor-pointer"
               />
diff --git a/enjoy/src/renderer/context/media-player-provider.tsx b/enjoy/src/renderer/context/media-player-provider.tsx
index e9deaee3..cb841200 100644
--- a/enjoy/src/renderer/context/media-player-provider.tsx
+++ b/enjoy/src/renderer/context/media-player-provider.tsx
@@ -142,6 +142,7 @@ export const MediaPlayerProvider = ({
   } = useRecordings(media, currentSegmentIndex);
 
   const initializeWavesurfer = async () => {
+    if (!layout?.playerHeight) return;
     if (!media) return;
     if (!mediaProvider) return;
     if (!ref?.current) return;
@@ -464,17 +465,12 @@ export const MediaPlayerProvider = ({
    * and mediaProvider is available
    */
   useEffect(() => {
-    if (!layout?.playerHeight) return;
-    if (!media) return;
-    if (!ref) return;
-    if (!mediaProvider) return;
-
     initializeWavesurfer();
-    setDecoded(false);
-    setDecodeError(null);
 
     return () => {
       if (wavesurfer) wavesurfer.destroy();
+      setDecoded(false);
+      setDecodeError(null);
     }
   }, [media, ref, mediaProvider, layout?.playerHeight]);