diff --git a/install.sh b/install.sh
index 08e9fd5..5591390 100644
--- a/install.sh
+++ b/install.sh
@@ -141,11 +141,12 @@ else
     echo "apt-transport-https is already installed."
 fi
 
-#APT Add contrib non-free to sources.list
-#sudo sed -i 's/^deb.* main/& contrib non-free/g' /etc/apt/sources.list
-
-#sudo sed -i 's/^Components:* main/& contrib non-free non-free-firmware/g' /etc/apt/sources.list.d/debian.sources
-
+# APT Add "contrib non-free" to the sources list
+if [ -f /etc/apt/sources.list.d/debian.sources ]; then
+    sudo sed -i 's/^Components:* main/& contrib non-free non-free-firmware/g' /etc/apt/sources.list.d/debian.sources
+else 
+	sudo sed -i 's/^deb.* main/& contrib non-free/g' /etc/apt/sources.list
+fi
 
 clear
 sudo apt update