wallpaper: Swap to os.environ.get

1 files changed, 4 insertions, 4 deletions

diff --git a/pywal/wallpaper.py b/pywal/wallpaper.py
index eef9a23..6cd70b6 100644
--- a/pywal/wallpaper.py
+++ b/pywal/wallpaper.py
@@ -8,19 +8,19 @@ from pywal import util
 
 def get_desktop_env():
     """Identify the current running desktop environment."""
-    desktop = os.getenv("XDG_CURRENT_DESKTOP")
+    desktop = os.environ.get("XDG_CURRENT_DESKTOP")
     if desktop:
         return desktop
 
-    desktop = os.getenv("DESKTOP_SESSION")
+    desktop = os.environ.get("DESKTOP_SESSION")
     if desktop:
         return desktop
 
-    desktop = os.getenv("GNOME_DESKTOP_SESSION_ID")
+    desktop = os.environ.get("GNOME_DESKTOP_SESSION_ID")
     if desktop:
         return "GNOME"
 
-    desktop = os.getenv("MATE_DESKTOP_SESSION_ID")
+    desktop = os.environ.get("MATE_DESKTOP_SESSION_ID")
     if desktop:
         return "MATE"