diff --git a/BUILD.bazel b/BUILD.bazel
index e1ebc19bd022..e338eb691f14 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -18,6 +18,7 @@ load(
     "checkpatch",
     "ddk_headers",
     "ddk_headers_archive",
+    "initramfs",
     "kernel_build",
     "kernel_compile_commands",
     "kernel_images",
@@ -972,18 +973,17 @@ kernel_modules_install(
     kernel_build = ":rockpi4",
 )
 
-kernel_images(
-    name = "rockpi4_images",
-    build_initramfs = True,
-    kernel_build = ":rockpi4",
+initramfs(
+    name = "rockpi4_initramfs",
     kernel_modules_install = ":rockpi4_modules_install",
+    visibility = ["//visibility:private"],
 )
 
 pkg_files(
     name = "rockpi4_dist_files",
     srcs = [
         ":rockpi4",
-        ":rockpi4_images",
+        ":rockpi4_initramfs",
         ":rockpi4_modules_install",
     ],
     strip_prefix = strip_prefix.files_only(),