diff --git a/home/gui/firefox.nix b/home/gui/firefox.nix
index a79cfd5..d5513a0 100644
--- a/home/gui/firefox.nix
+++ b/home/gui/firefox.nix
@@ -13,11 +13,16 @@ in with lib;
 {
 	programs.firefox = {
 		enable = true;
-		enableGnomeExtensions = true;
+		package = pkgs.firefox-wayland.override {
+			cfg = {
+				enableGnomeExtensions = true;
+			};
+		};
 		extensions = with pkgs.nur.repos.rycee.firefox-addons; [
 			keepassxc-browser
 			octotree
 			refined-github
+			tree-style-tab
 			ublock-origin
 		];
 		profiles = {
diff --git a/home/home.nix b/home/home.nix
index 8ecf54a..62cd5b3 100644
--- a/home/home.nix
+++ b/home/home.nix
@@ -18,4 +18,5 @@ in {
 
 	home.username = name;
 	home.homeDirectory = if name == "root" then "/root" else "/home/${name}";
+	home.stateVersion = "21.11";
 }