diff --git a/sh_scripts/official_repo.sh b/sh_scripts/official_repo.sh
index 3611613..2e872c9 100644
--- a/sh_scripts/official_repo.sh
+++ b/sh_scripts/official_repo.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-SYSTEM=$(lsb_release -sr)
+SYSTEM=$(lsb_release -sr | cut -d. -f1)
 # Make sure only root can run our script
 if [ "$(id -u)" != "0" ]; then
    echo "This script must be run as root" 1>&2
@@ -42,25 +42,25 @@ if [ "$SYSTEM" == "10" ]; then
     apt-get install curl wget apt-transport-https dirmngr neofetch htop nload git -y
     echo "Have Fun with the Official Repos :)"
 fi
-if [ "$SYSTEM" == "9" ]; then
+if [ "$OLD" == "9" ]; then
     curl "https://raw.githubusercontent.com/MrUnknownDE/linux-script/main/sh_scripts/repos/debian-9.txt" -o /etc/apt/sources.list
     apt-get update && apt-get upgrade -y
     apt-get install curl wget apt-transport-https dirmngr neofetch htop nload git -y
     echo "Have Fun with the Official Repos :)"
 fi
-if [ "$SYSTEM" == "20.04" ]; then
+if [ "$SYSTEM" == "20" ]; then
     curl "https://raw.githubusercontent.com/MrUnknownDE/linux-script/main/sh_scripts/repos/ubuntu-20.04.txt" -o /etc/apt/sources.list
     apt-get update && apt-get upgrade -y
     apt-get install curl wget apt-transport-https dirmngr neofetch htop nload git -y
     echo "Have Fun with the Official Repos :)"
 fi
-if [ "$SYSTEM" == "18.04" ]; then
+if [ "$SYSTEM" == "18" ]; then
     curl "https://raw.githubusercontent.com/MrUnknownDE/linux-script/main/sh_scripts/repos/ubuntu-18.04.txt" -o /etc/apt/sources.list
     apt-get update && apt-get upgrade -y
     apt-get install curl wget apt-transport-https dirmngr neofetch htop nload git -y
     echo "Have Fun with the Official Repos :)"
 fi
-if [ "$SYSTEM" ==  "16.04" ]; then
+if [ "$SYSTEM" ==  "16" ]; then
     curl "https://raw.githubusercontent.com/MrUnknownDE/linux-script/main/sh_scripts/repos/ubuntu-16.04.txt" -o /etc/apt/sources.list
     apt-get update && apt-get upgrade -y
     apt-get install curl wget apt-transport-https dirmngr neofetch htop nload git -y