diff --git a/hosts/nixos-desktop/default.nix b/hosts/nixos-desktop/default.nix
index bed3eca..84d74d2 100644
--- a/hosts/nixos-desktop/default.nix
+++ b/hosts/nixos-desktop/default.nix
@@ -29,8 +29,6 @@
     services.wallpaper.enable = true;
   };
 
-  virtualisation.containers.cdi.dynamic.nvidia.enable = true;
-
   home-manager.users.moritz.home.packages = with pkgs; [
     anki
     stable.calibre # NOTE: breaks often in unstable
@@ -38,10 +36,14 @@
 
   hardware = {
     keyboard.qmk.enable = true;
-    nvidia.modesetting.enable = true;
-    opengl = {
+    nvidia = {
+      modesetting.enable = true;
+      open = true;
+    };
+    nvidia-container-toolkit.enable = true;
+    graphics = {
       enable = true;
-      driSupport32Bit = true;
+      enable32Bit = true;
     };
 
     # sensors
@@ -50,9 +52,6 @@
   };
 
   boot = {
-    # KERNEL
-    kernelPackages = config.boot.zfs.package.latestCompatibleLinuxPackages; # NOTE: use latest zfs compatible kernel
-
     # BOOT
     supportedFilesystems = [ "zfs" "btrfs" "ntfs" ];
     loader = {
diff --git a/hosts/nixos-desktop/disko.nix b/hosts/nixos-desktop/disko.nix
index d373c43..43d1242 100644
--- a/hosts/nixos-desktop/disko.nix
+++ b/hosts/nixos-desktop/disko.nix
@@ -1,4 +1,4 @@
-{ lib, ... }:
+{ lib, pkgs, ... }:
 
 {
   # needed for zfs pool
@@ -83,9 +83,27 @@
     };
   };
   # rollback to blank
-  boot.initrd.postDeviceCommands = lib.mkAfter ''
-    zfs rollback -r zroot/encrypted/root@blank && echo "rollback complete"
-  '';
+  boot.initrd.systemd.services.rollback = {
+    description = "Rollback ZFS datasets to a pristine state";
+    wantedBy = [
+      "initrd.target"
+    ];
+    after = [
+      "zfs-import-zroot.service"
+    ];
+    before = [
+      "sysroot.mount"
+    ];
+    path = with pkgs; [
+      zfs
+    ];
+    unitConfig.DefaultDependencies = "no";
+    serviceConfig.Type = "oneshot";
+    script = ''
+      zfs rollback -r zroot/encrypted/root@blank && echo "rollback complete"
+    '';
+  };
+  boot.initrd.systemd.enable = true;
   fileSystems."/persist".neededForBoot = true;
   # HACK: to fix issue of agenix running before impermanence
   age.identityPaths = [
diff --git a/modules/profiles/base.nix b/modules/profiles/base.nix
index 7586fad..ca8a8b4 100644
--- a/modules/profiles/base.nix
+++ b/modules/profiles/base.nix
@@ -241,15 +241,16 @@ in
     which-nix
 
     # utils
+    (viu.override { withSixel = true; })
     bat
     cht-sh
     f
     fd
     gi
+    lazygit
     parallel
     ripgrep
     vim
-    (viu.override { withSixel = true; })
     wget
   ];
 
diff --git a/modules/programs/nvim/new_plugins/lazygit.nix b/modules/programs/nvim/new_plugins/lazygit.nix
new file mode 100644
index 0000000..9cb3060
--- /dev/null
+++ b/modules/programs/nvim/new_plugins/lazygit.nix
@@ -0,0 +1,14 @@
+_:
+
+{
+  home-manager.users.moritz.programs.nixvim = {
+    keymaps = [
+      { key = "<leader>g"; action = "<cmd>LazyGit<cr>"; options.desc = "Lazygit"; }
+    ];
+
+    plugins.lazygit = {
+      enable = true;
+      settings = { };
+    };
+  };
+}