diff --git a/flake.nix b/flake.nix
index a229724..a8d0a90 100644
--- a/flake.nix
+++ b/flake.nix
@@ -94,8 +94,10 @@
                 })
               ];
               kernelParams = [
+                # TODO: Use the scripts instead https://github.com/tenstorrent/tt-system-tools they work regardless of how many devices there are
+                # https://github.com/tenstorrent/tt-system-tools/blob/main/hugepages-setup.sh#L31
                 "hugepagesz=1G"
-                "hugepages=2"
+                "hugepages=4"
                 "iommu=pt"
               ];
               kernelModules = [ "tenstorrent" ];