diff --git a/main.go b/main.go
index e9cb799..a5313e6 100644
--- a/main.go
+++ b/main.go
@@ -512,7 +512,7 @@ func getTerminal() string {
 }
 
 func getDE() string {
-	de := os.Getenv("DESKTOP_SESSION")
+	de := os.Getenv("XDG_CURRENT_DESKTOP")
 	if de == "" {
 		return "Unknown or w/o GUI"
 	}
@@ -543,7 +543,7 @@ func main() {
 	conf, _ := initConfig()
 
 	if len(conf.Layout) == 0 {
-		conf.Layout = []string{"host","dist", "cpu", "ram", "shell", "uptime"}
+		conf.Layout = []string{"host", "dist", "cpu", "ram", "shell", "uptime"}
 		fmt.Println("Warning: No layout specified in config, using default")
 	}
 
@@ -573,7 +573,7 @@ func main() {
 	for _, item := range conf.Layout {
 		switch strings.ToLower(item) {
 		case "host":
-				stats = append(stats, Info{"host", getHostname()})
+			stats = append(stats, Info{"host", getHostname()})
 		case "dist":
 			stats = append(stats, Info{"OS", dist})
 		case "cpu":