diff --git a/battery/build b/battery/build
index c44922a..1d99e2b 100755
--- a/battery/build
+++ b/battery/build
@@ -1,6 +1,7 @@
 #!/usr/bin/bash
 
 HEADERS=6.12.47+rpt-rpi-v8
+HEADERS=$(uname -r)
 make -C /lib/modules/$HEADERS/build M=$PWD clean
 make -C /lib/modules/$HEADERS/build M=$PWD modules