diff --git a/modules/gui/firefox.nix b/modules/gui/firefox.nix
index f9610b3..cb7b4c2 100644
--- a/modules/gui/firefox.nix
+++ b/modules/gui/firefox.nix
@@ -72,11 +72,13 @@ in {
         };
 
         home-manager.users.${username} = {
-            xdg.desktopEntries = {
+            xdg.desktopEntries = let
+                pkg = config.home-manager.users.${username}.programs.firefox.package;
+            in {
                 firefox-schizo = {
                     name = "Firefox Schizo";
                     genericName = "Web Browser";
-                    exec = "${getExe pkgs.firefox-devedition-bin} --name firefox-schizo -P schizo %U";
+                    exec = "${getExe pkg} --name firefox-schizo -P schizo %U";
                     icon = "${logo}";
                     terminal = false;
                     categories = [ "Network" "WebBrowser" ];
@@ -90,7 +92,7 @@ in {
                 firefox-unschizo = {
                     name = "Firefox Unschizo";
                     genericName = "Web Browser";
-                    exec = "${getExe pkgs.firefox-devedition-bin} --name firefox-unschizo -P unschizo %U";
+                    exec = "${getExe pkg} --name firefox-unschizo -P unschizo %U";
                     icon = "${logo}";
                     terminal = false;
                     categories = [ "Network" "WebBrowser" ];