diff --git a/environment.py b/environment.py
index 12fda05..bb7fbf6 100644
--- a/environment.py
+++ b/environment.py
@@ -53,7 +53,7 @@ def detect_environment():
     This sets wallpapers on any platform, as long as it is supported.
 """
 def set_environment_wallpapers(environment: str, files: list, displays: list):
-    if environment in environment_handlers:
+    if f"{environment}_setter" in environment_handlers:
         environment_handlers[f"{environment}_setter"](files, displays)
         logging.info("Wallpapers set!")
     else:
diff --git a/main.py b/main.py
index b8d3fdf..5110b13 100644
--- a/main.py
+++ b/main.py
@@ -47,7 +47,7 @@ def main():
     if not args.environment:
         set_environment_wallpapers(environment, files, displays)
     else:
-        environment_handlers[args.environment](files, displays)
+        environment_handlers[f"{args.environment}_setter"](files, displays)
         logging.info("Wallpapers set!")