From a249ef27331f88b1631e51e2736decb0d2babdb3 Mon Sep 17 00:00:00 2001
From: Skullheadx <94652084+Skullheadx@users.noreply.github.com>
Date: Sun, 13 Oct 2024 23:07:02 -0400
Subject: [PATCH] add fix for downloading youtu.be links

---
 ytdl/__main__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ytdl/__main__.py b/ytdl/__main__.py
index 6a653b1..0a6919e 100644
--- a/ytdl/__main__.py
+++ b/ytdl/__main__.py
@@ -14,7 +14,7 @@ def main():
         if arg in modes:
             mode = arg
 
-        if "youtube.com" in arg:
+        if "youtube" in arg or "youtu.be" in arg:
             links.extend(arg.split(" "))
 
     assert len(links) > 0, "Should pass at least one link as arg"
-- 
2.54.0