diff --git a/lovelace/04_media.yaml b/lovelace/04_media.yaml
index b7628e1..13522e5 100644
--- a/lovelace/04_media.yaml
+++ b/lovelace/04_media.yaml
@@ -15,6 +15,7 @@ cards:
         type: "custom:text-input-row"
 
   - type: entity-button
+    entity: script.youtube_download_selected
     name: Download
     icon: mdi:download
     icon_height: 48px