diff --git a/Installation/arch.sh b/Installation/arch.sh
index f61d79a..e381064 100644
--- a/Installation/arch.sh
+++ b/Installation/arch.sh
@@ -124,8 +124,10 @@ install_flat() {
 # Entry Point
 main() {
     ## Desktop Specific Operations ##
-    if [ "$DESKTOP_SESSION" == "gnome" ]; then
+    if [ "$XDG_CURRENT_DESKTOP" == "Gnome" ]; then
         gnome
+    elif [ "$XDG_CURRENT_DESKTOP" == "KDE" ]; then
+        kde
     fi
 
     # Installing Software