diff --git a/server_files/model.php b/server_files/model.php
index f89fcf6..263dcc8 100644
--- a/server_files/model.php
+++ b/server_files/model.php
@@ -298,8 +298,8 @@ function exportProblemsData() {
 
 function displayVideo($link)
 {
-	if (preg_match("#(https?://|)(www\.|)youtube\.com/watch\?v=(.*)#", $link, $matches)) {
-		$code = $matches[3];
+	if (preg_match("#(https?://|)(www\.|)(youtube\.com/watch\?v=|youtu.be/)(.*)#", $link, $matches)) {
+		$code = $matches[4];
         /** @noinspection HtmlDeprecatedAttribute */
         /** @noinspection HtmlDeprecatedTag */
         echo "<center><iframe width=\"854px\" height=\"480px\" src=\"https://www.youtube.com/embed/$code\" frameborder=\"0\" allow=\"accelerometer; autoplay; encrypted-media; gyroscope; picture-in-picture\" allowfullscreen></iframe></center><br />\n";