diff --git a/TODO b/TODO
index 8ec63ac..4ad5216 100644
--- a/TODO
+++ b/TODO
@@ -1,2 +1 @@
 - test dkms fallback method
-  - match zfs-linux-dkms with current running kernel : http://sprunge.us/kPWTJf
diff --git a/init b/init
index 3b65dbd..1da8a92 100755
--- a/init
+++ b/init
@@ -141,7 +141,7 @@ dkms_init () {
     pacman -Syyuu --noconfirm
 
     print "Install base-devel"
-    pacman -S --noconfirm base-devel
+    pacman -S --noconfirm base-devel linux-headers
 
 }
 
@@ -198,7 +198,7 @@ else
 
     # DKMS fallback
     print "No zfs-linux package was found for current running kernel, fallback on DKMS method"
-    dkms_init "$kernel_version_fixed_for_headers"
+    dkms_init
 
     print "Install zfs-dkms"