diff --git a/src/timelines/MediaAttachmentGrid.tsx b/src/timelines/MediaAttachmentGrid.tsx
index b0703e0..625e0ab 100644
--- a/src/timelines/MediaAttachmentGrid.tsx
+++ b/src/timelines/MediaAttachmentGrid.tsx
@@ -152,6 +152,7 @@ const MediaAttachmentGrid: Component<{
               return (
                 <img
                   data-sort={index}
+                  data-media-type={item().type}
                   src={item().previewUrl}
                   width={item().meta?.small?.width}
                   height={item().meta?.small?.height}
@@ -165,6 +166,7 @@ const MediaAttachmentGrid: Component<{
               return (
                 <video
                   data-sort={index}
+                  data-media-type={item().type}
                   src={item().url || undefined}
                   autoplay={settings().autoPlayVideos}
                   playsinline={settings().autoPlayVideos ? true : undefined}
@@ -178,6 +180,8 @@ const MediaAttachmentGrid: Component<{
             case "gifv":
               return (
                 <video
+                  data-sort={index}
+                  data-media-type={item().type}
                   src={item().url || undefined}
                   autoplay={settings().autoPlayGIFs}
                   controls
@@ -191,6 +195,14 @@ const MediaAttachmentGrid: Component<{
               );
 
             case "audio":
+              return (
+                <audio
+                  data-sort={index}
+                  data-media-type={item().type}
+                  src={item().url || undefined}
+                  controls
+                ></audio>
+              );
             case "unknown":
               return <div></div>;
           }