diff --git a/modules/nixos/system/default.nix b/modules/nixos/system/default.nix
index a2bcd3f..521a547 100644
--- a/modules/nixos/system/default.nix
+++ b/modules/nixos/system/default.nix
@@ -36,6 +36,7 @@ in
           home-manager
           openssh
           sudo
+          treefmt2
           xz.bin
           # Packages required for decrypting config files
           transcrypt
diff --git a/systems/x86_64-linux/Hevana/hardware-configuration.nix b/systems/x86_64-linux/Hevana/hardware-configuration.nix
index 48f9011..aa78f73 100644
--- a/systems/x86_64-linux/Hevana/hardware-configuration.nix
+++ b/systems/x86_64-linux/Hevana/hardware-configuration.nix
@@ -1,5 +1,10 @@
 # Minisforum UM340
-{ modulesPath, namespace, ... }:
+{
+  modulesPath,
+  namespace,
+  pkgs,
+  ...
+}:
 let
   bootUUID = "D2E7-FE8F"; # The UUID of the boot partition.
   luksUUID = "7b9c756c-ba9d-43fc-b935-7c77a70f5f1b"; # The UUID of the locked LUKS partition.
@@ -8,6 +13,7 @@ in
   imports = [ (modulesPath + "/installer/scan/not-detected.nix") ];
 
   boot = {
+    kernelPackages = pkgs.linuxKernel.packages.linux_zen;
     kernelModules = [ "kvm-amd" ];
 
     initrd.kernelModules = [