diff --git a/setup-nasbeery b/setup-nasbeery
index 196027f..68fa14f 100644
--- a/setup-nasbeery
+++ b/setup-nasbeery
@@ -106,6 +106,8 @@ DEBIAN_FRONTEND=noninteractive DEBIAN_PRIORITY=critical apt -y -qq -o DPkg::opti
 echo "Detecting Architecture"
 if [[ $(dpkg --get-selections | grep -m1 "raspberrypi-kernel") ]]; then
 	headers="raspberrypi-kernel-headers"
+elif [[ $(dpkg --get-selections | grep -m1 "linux-image-raspi") ]]; then
+  headers="linux-headers-raspi"
 elif [[ $(dpkg --get-selections | grep -E -m1 "linux-image-current-rockchip64") ]]; then
 	headers="linux-headers-current-rockchip64"
 elif [[ $(dpkg --get-selections | grep -E -m1 "linux-image-edge-rockchip64") ]]; then