add fix for downloading youtu.be links

diff --git a/ytdl/__main__.py b/ytdl/__main__.py
index 6a653b1f4f15f03b539961390c8b9171be4b39b1..0a6919ee2a737ab8f63d551735c9a887c9403e4c 100644 (file)
--- 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"