diff --git a/hardware-configuration.nix b/hardware-configuration.nix
index e004655..6aa649f 100644
--- a/hardware-configuration.nix
+++ b/hardware-configuration.nix
@@ -8,8 +8,7 @@
 
   boot = {
     tmp.cleanOnBoot = true;
-    kernelModules = [ "cma=256M" ];
-    extraModulePackages = [ ];
+    kernelPackages = pkgs.linuxPackages_rpi;
 
     loader = {
       grub = {
@@ -17,17 +16,9 @@
         enable = false;
       };
 
-      raspberryPi = {
+      generic-extlinux-compatible = {
         enable = true;
-        version = 3;
-        firmwareConfig = ''
-          gpu_mem=256
-        '';
-
-        uboot = {
-          enable = true;
-          configurationLimit = 10;
-        };
+        configurationLimit = 10;
       };
     };
   };
@@ -57,6 +48,7 @@
   swapDevices = [
     {
       device = "/swapfile";
+      size = 1024;
     }
   ];