diff --git a/yt_summarizer.py b/yt_summarizer.py
index b34c821..65cf432 100644
--- a/yt_summarizer.py
+++ b/yt_summarizer.py
@@ -7,7 +7,7 @@ import re
 
 
 def check_link(link):
-    yt_regex = r"(https?://)?(www\.)?(youtube\.com/watch\?v=|youtu\.be/)[\w-]+"
+    yt_regex = r"^(?:https?:\/\/)?(?:www\.)?youtu(?:be\.com\/watch\?v=|\.be\/)([\w\-\_]+)(?:\?.*)?$"
     return re.match(yt_regex, link) is not None