diff --git a/flake.nix b/flake.nix
index 6d12f63..0456a93 100644
--- a/flake.nix
+++ b/flake.nix
@@ -25,7 +25,7 @@
           imports = [
               ./nixos.nix
               ./hardware-configuration/iota.nix
-              {_module.args = {inherit inputs;};}
+              { _module.args = { inherit inputs; }; }
           ];
           home-manager.users.steph = { imports = [ ./home.nix ./platforms/linux/home.nix ]; };
           home-manager.extraSpecialArgs = { inherit inputs ; os = "linux"; };
@@ -40,10 +40,10 @@
           imports = [
               ./nixos.nix
               ./hardware-configuration/tau.nix
-              {_module.args = {inherit inputs;};}
+              { _module.args = { inherit inputs; }; }
           ];
           home-manager.users.steph = { imports = [ ./home.nix ./platforms/linux/home.nix ]; };
-          home-manager.extraSpecialArgs = { inherit inputs ; os = "linux"; };
+          home-manager.extraSpecialArgs = { inherit inputs; os = "linux"; };
         }
       ];
     };
@@ -56,6 +56,5 @@
         mac-app-util.homeManagerModules.default 
       ];
     };
-  
   };
 }
diff --git a/programs/firefox.nix b/programs/firefox.nix
index 218945e..c399308 100644
--- a/programs/firefox.nix
+++ b/programs/firefox.nix
@@ -10,8 +10,7 @@
       inputs.firefox-macos-theme;
   })
 
-  {
-    programs.librewolf = lib.mkMerge [
+  (let configedFirefox = lib.mkMerge [
       (lib.mkIf (os == "darwin") {
         package = null;
       })
@@ -93,5 +92,13 @@
         };
       }
     ];
-  }
+  in lib.mkMerge [
+      (lib.mkIf (os == "linux") {
+        programs.librewolf = configedFirefox;
+      })
+
+      (lib.mkIf (os == "darwin") {
+        programs.firefox = configedFirefox;
+      })
+  ])
 ]