diff --git a/flake.nix b/flake.nix
index 58ec4fd..7c8e5dc 100644
--- a/flake.nix
+++ b/flake.nix
@@ -1,30 +1,22 @@
-# This flake was initially generated by fh, the CLI for FlakeHub (version 0.1.22)
 {
-  # A helpful description of your flake
   description = "Music Presence flake";
 
-  # Flake inputs
   inputs = {
     flake-schemas.url = "https://flakehub.com/f/DeterminateSystems/flake-schemas/*";
     nixpkgs.url = "https://flakehub.com/f/NixOS/nixpkgs/*";
   };
 
-  # Flake outputs that other flakes can use
   outputs = { self, flake-schemas, nixpkgs }:
     let
-      # Helpers for producing system-specific outputs
       supportedSystems = [ "x86_64-linux" ];
       forEachSupportedSystem = f: nixpkgs.lib.genAttrs supportedSystems (system: f {
         pkgs = import nixpkgs { inherit system; };
       });
     in {
-      # Schemas tell Nix about the structure of your flake's outputs
       schemas = flake-schemas.schemas;
 
-      # Development environments
       devShells = forEachSupportedSystem ({ pkgs }: {
         default = pkgs.mkShell {
-          # Pinned packages available in the environment
           packages = with pkgs; [
             curl
             git
@@ -42,20 +34,20 @@
 
           src = pkgs.fetchurl {
             url = "https://github.com/ungive/discord-music-presence/releases/download/v${version}/${pname}-${version}-linux-x86_64.AppImage";
-            hash = "";
+            hash = "sha256-S8VHs81wbVv5Z1lncSIYJDUHRfUBl40WU+P3QTQROp0=";
           };
 
           appimageContents = pkgs.appimageTools.extract { inherit pname version src; };
-
         in pkgs.appimageTools.wrapType2 {
           inherit pname version src;
-          
+
+          nativeBuildInputs = with pkgs; [ makeWrapper ];
           extraInstallCommands = ''
-            install -m 444 -D ${appimageContents}/${pname}.desktop -t $out/share/applications
-            substituteInPlace $out/share/applications/${pname}.desktop \
-              --replace-fail 'Exec=AppRun' 'Exec=${pname}'
-            cp -r ${appimageContents}/usr/share/icons $out/share
-          '';
+        wrapProgram $out/bin/${pname} \
+          --set QT_QPA_PLATFORM xcb \
+          --add-flags "\''${NIXOS_OZONE_WL:+\''${WAYLAND_DISPLAY:+--ozone-platform-hint=auto --enable-features=WaylandWindowDecorations}}"
+        install -Dm444 ${appimageContents}/musicpresence.desktop -t $out/share/applications
+      '';
 
           passthru.updateScript = pkgs.nix-update-script { };