diff --git a/modules/vscode.nix b/modules/vscode.nix
index 2bde050..cef327e 100644
--- a/modules/vscode.nix
+++ b/modules/vscode.nix
@@ -1,4 +1,5 @@
 { pkgs
+, username
 , vscode-extensions
 , system
 , home
@@ -6,19 +7,23 @@
 }: {
   home.programs.vscode = {
     enable = true;
-    package = pkgs.vscode-with-extensions.override {
-      vscodeExtensions = with vscode-extensions.packages.${system}; [
-        vscode.quandinh.onehalf-dark
-        pkgs.vscode-extensions.ms-vsliveshare.vsliveshare
-        pkgs.vscode-extensions.asvetliakov.vscode-neovim
-        pkgs.vscode-extensions.ms-vscode-remote.remote-ssh
-      ];
-    } // {
-      pname = "vscode";
-    };
-    mutableExtensionsDir = false;
+    enableUpdateCheck = false;
+    enableExtensionUpdateCheck = false;
+    package = pkgs.vscode-fhs;
+    extensions = with vscode-extensions.packages.${system}; [
+      vscode.quandinh.onehalf-dark
+      pkgs.vscode-extensions.ms-vsliveshare.vsliveshare
+      pkgs.vscode-extensions.ms-vscode-remote.remote-ssh
+    ] ++ pkgs.vscode-utils.extensionsFromVscodeMarketplace [
+      /* { */
+      /*   name = "codeium"; */
+      /*   publisher = "Codeium"; */
+      /*   version = "1.1.51"; */
+      /*   sha256 = "sha256-MgIRItR2QhGk9U2x+nWjOkUYJxEwYzaKOsxfptpVDaw="; */
+      /* } */
+    ];
+    mutableExtensionsDir = true;
     userSettings = {
-      "update.channel" = "none";
       "workbench.colorTheme" = "onehalf-dark";
       "workbench.startupEditor" = "none";
       "files.autoSave" = "afterDelay";
@@ -29,6 +34,7 @@
       };
       "vscode-neovim.neovimExecutablePaths.linux" = "${home.programs.neovim.finalPackage}/bin/nvim";
       "security.workspace.trust.enabled" = false;
+      "codeium.enableSearch" = true;
     };
   };
   config.internal.allowUnfree = [
@@ -36,5 +42,7 @@
     "vscode-extension-ms-vscode-remote-remote-ssh"
     "vscode"
     "vscode-with-extensions"
+    "code"
   ];
+  config.environment.persistence."/large/persist".users.${username}.directories = [ ".vscode/extensions" ];
 }