diff --git a/PKGBUILD b/PKGBUILD
index 0e88e4c..7d808f8 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -40,11 +40,8 @@ package() {
     # Install binary
     install -Dm755 system76-power-gui "$pkgdir/usr/bin/system76-power-gui"
     
-    # Create appropriate desktop entry based on system
-    if [ -n "${WAYLAND_DISPLAY}" ] || [ "${XDG_SESSION_TYPE}" = "wayland" ] || [ -n "${WAYLAND_SOCKET}" ]; then
-        # Install Wayland desktop entry
-        install -Dm644 system76-power-gui.desktop "$pkgdir/usr/share/applications/system76-power-gui.desktop"
-    fi
+    # Install for Wayland
+    install -Dm644 system76-power-gui.desktop "$pkgdir/usr/share/applications/system76-power-gui.desktop"
     
     # Install polkit rules
     install -Dm644 90-system76-power.rules "$pkgdir/usr/share/polkit-1/rules.d/90-system76-power.rules"