diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index eeac63b3..fc219fde 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -45,7 +45,7 @@ jobs:
         run: sudo apt update -y
 
       - name: Install build dependencies
-        run: sudo DEBIAN_FRONTEND=noninteractive apt install libssl-dev libyaml-dev libc-ares-dev libprotobuf-dev protobuf-compiler libjq-dev libyaml-cpp-dev libgrpc++-dev protobuf-compiler-grpc rpm libelf-dev cmake build-essential libcurl4-openssl-dev linux-headers-generic clang llvm git -y
+        run: sudo DEBIAN_FRONTEND=noninteractive apt install libssl-dev libyaml-dev libc-ares-dev libprotobuf-dev protobuf-compiler libjq-dev libyaml-cpp-dev libgrpc++-dev protobuf-compiler-grpc rpm libelf-dev cmake build-essential libcurl4-openssl-dev linux-headers-$(uname -r) clang llvm git -y
 
       - name: Prepare project
         run: |
@@ -57,7 +57,7 @@ jobs:
       - name: Build
         run: |
           pushd build
-          KERNELDIR=/lib/modules/$(ls /lib/modules)/build make -j4 all
+          KERNELDIR=/lib/modules/$(uname -r)/build make -j4 all
           popd
 
       - name: Run unit tests
@@ -76,7 +76,7 @@ jobs:
         run: sudo apt update -y
 
       - name: Install build dependencies
-        run: sudo DEBIAN_FRONTEND=noninteractive apt install libssl-dev libyaml-dev libc-ares-dev libprotobuf-dev protobuf-compiler libjq-dev libyaml-cpp-dev libgrpc++-dev protobuf-compiler-grpc rpm libelf-dev cmake build-essential libcurl4-openssl-dev linux-headers-generic clang llvm git -y
+        run: sudo DEBIAN_FRONTEND=noninteractive apt install libssl-dev libyaml-dev libc-ares-dev libprotobuf-dev protobuf-compiler libjq-dev libyaml-cpp-dev libgrpc++-dev protobuf-compiler-grpc rpm libelf-dev cmake build-essential libcurl4-openssl-dev linux-headers-$(uname -r) clang llvm git -y
 
       - name: Prepare project
         run: |
@@ -88,7 +88,7 @@ jobs:
       - name: Build
         run: |
           pushd build
-          KERNELDIR=/lib/modules/$(ls /lib/modules)/build make -j4 all
+          KERNELDIR=/lib/modules/$(uname -r)/build make -j4 all
           popd
 
       - name: Run unit tests
@@ -107,7 +107,7 @@ jobs:
         run: sudo apt update -y
 
       - name: Install build dependencies
-        run: sudo DEBIAN_FRONTEND=noninteractive apt install cmake build-essential clang llvm git linux-headers-generic pkg-config autoconf libtool libelf-dev -y
+        run: sudo DEBIAN_FRONTEND=noninteractive apt install cmake build-essential clang llvm git linux-headers-$(uname -r) pkg-config autoconf libtool libelf-dev -y
 
       - name: Prepare project
         run: |
@@ -119,7 +119,7 @@ jobs:
       - name: Build
         run: |
           pushd build
-          KERNELDIR=/lib/modules/$(ls /lib/modules)/build make -j4 all
+          KERNELDIR=/lib/modules/$(uname -r)/build make -j4 all
           popd
 
       - name: Run unit tests