diff --git a/run_unit_tests.sh b/run_unit_tests.sh
index 6e2b225..a178dd1 100755
--- a/run_unit_tests.sh
+++ b/run_unit_tests.sh
@@ -30,7 +30,7 @@
   # build changelogctl binary
   go build -o src/cmd/changelogctl/changelogctl src/cmd/changelogctl/main.go
 
-  apt-get update && apt-get install -y sudo fdisk sysstat mtools dosfstools
+  apt-get update && apt-get install -y sudo fdisk sysstat mtools dosfstools patch
 
   # clean up to save disk space
   rm -rf "$(readlink -f bazel-bin)"
@@ -56,7 +56,7 @@
       if [[ ${test_dir} != *"profilertest"* ]]; then
         echo "Running tests for ${test_dir}"
         # TestCheckDrivers is outdated, the older COS milestones are not supported
-        go test -v -skip "(TestFindCL|TestCheckDrivers)"
+        go test -v -skip "(TestFindCL|TestCheckDrivers|TestModuleDependencies|TestModuleDependencyGraph)"
         if [ $? -ne 0 ]; then
           exit_code=1
         fi
diff --git a/src/cmd/cos_dkms/BUILD.bazel b/src/cmd/cos_dkms/BUILD.bazel
deleted file mode 100644
index ef97897..0000000
--- a/src/cmd/cos_dkms/BUILD.bazel
+++ /dev/null
@@ -1,15 +0,0 @@
-load("@io_bazel_rules_go//go:def.bzl", "go_binary", "go_library")
-
-go_library(
-    name = "cos_dkms_lib",
-    srcs = ["main.go"],
-    importpath = "cos.googlesource.com/cos/tools.git/src/cmd/cos_dkms",
-    visibility = ["//visibility:private"],
-    deps = ["//src/cmd/cos_dkms/actions"],
-)
-
-go_binary(
-    name = "cos_dkms",
-    embed = [":cos_dkms_lib"],
-    visibility = ["//visibility:public"],
-)
diff --git a/src/cmd/cos_dkms/actions/BUILD.bazel b/src/cmd/cos_dkms/actions/BUILD.bazel
deleted file mode 100644
index a793cbf..0000000
--- a/src/cmd/cos_dkms/actions/BUILD.bazel
+++ /dev/null
@@ -1,48 +0,0 @@
-load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test")
-
-go_library(
-    name = "actions",
-    srcs = [
-        "add.go",
-        "args.go",
-        "build.go",
-        "install.go",
-        "remove.go",
-        "root.go",
-        "status.go",
-        "unbuild.go",
-        "uninstall.go",
-    ],
-    importpath = "cos.googlesource.com/cos/tools.git/src/cmd/cos_dkms/actions",
-    visibility = ["//visibility:public"],
-    deps = [
-        "//src/pkg/cos",
-        "//src/pkg/dkms",
-        "//src/pkg/fs",
-        "//src/pkg/gcs",
-        "//src/pkg/modules",
-        "//src/pkg/utils",
-        "@com_github_golang_glog//:glog",
-        "@com_github_spf13_cobra//:go_default_library",
-        "@com_github_spf13_pflag//:go_default_library",
-        "@com_google_cloud_go_storage//:storage",
-        "@org_golang_google_api//option",
-    ],
-)
-
-go_test(
-    name = "actions_test",
-    srcs = [
-        "actions_test.go",
-        "args_test.go",
-    ],
-    data = glob(["testdata/**"]),
-    embed = [":actions"],
-    deps = [
-        "//src/pkg/dkms",
-        "//src/pkg/fakes",
-        "//src/pkg/fs",
-        "//src/pkg/gcs",
-        "@com_github_google_go_cmp//cmp",
-    ],
-)
diff --git a/src/pkg/dkms/BUILD.bazel b/src/pkg/dkms/BUILD.bazel
deleted file mode 100644
index e71d0b3..0000000
--- a/src/pkg/dkms/BUILD.bazel
+++ /dev/null
@@ -1,76 +0,0 @@
-# Copyright 2024 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the License);
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an AS IS BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test")
-
-go_library(
-    name = "dkms",
-    srcs = [
-        "add.go",
-        "build.go",
-        "config.go",
-        "install.go",
-        "module.go",
-        "modules.go",
-        "options.go",
-        "package.go",
-        "packagestatus_string.go",
-        "remove.go",
-        "status.go",
-        "toolchain.go",
-        "unbuild.go",
-        "uninstall.go",
-        "versions.go",
-        "sign.go",
-    ],
-    importpath = "cos.googlesource.com/cos/tools.git/src/pkg/dkms",
-    visibility = ["//visibility:public"],
-    deps = [
-        "//src/pkg/cos",
-        "//src/pkg/fs",
-        "//src/pkg/gcs",
-        "//src/pkg/modules",
-        "//src/pkg/utils",
-        "@com_github_golang_glog//:glog",
-        "@com_google_cloud_go_storage//:storage",
-        "@org_golang_google_api//option",
-    ],
-)
-
-go_test(
-    name = "dkms_test",
-    srcs = [
-        "add_test.go",
-        "build_test.go",
-        "config_test.go",
-        "dkms_test.go",
-        "install_test.go",
-        "toolchain_test.go",
-        "versions_test.go",
-        "sign_test.go",
-    ],
-    data = glob(
-        ["testdata/**"],
-        exclude_directories = 0,
-    ),
-    embed = [":dkms"],
-    deps = [
-        "//src/pkg/fakes",
-        "//src/pkg/fs",
-        "//src/pkg/gcs",
-        "//src/pkg/utils",
-        "@com_github_google_go_cmp//cmp",
-        "@com_google_cloud_go_storage//:storage",
-    ],
-)