diff --git a/data/data.go b/data/data.go
index 5a502e7..052170a 100644
--- a/data/data.go
+++ b/data/data.go
@@ -158,7 +158,10 @@ func GetShell() string {
 	return os.Getenv("SHELL")
 }
 func GetWM() string {
-	return os.Getenv("XDG_CURRENT_DESKTOP")
+	cmd := exec.Command("wmctrl", "-m")
+	shell, _ := cmd.Output()
+	line := strings.Split(string(shell), "\n")[0]
+	return strings.ReplaceAll(line, "Name: ", "")
 }
 func getPackages() {
 }