diff --git a/cloud/agent/api/BUILD.bazel b/cloud/agent/api/BUILD.bazel
index 3742cf5..9312b9d 100644
--- a/cloud/agent/api/BUILD.bazel
+++ b/cloud/agent/api/BUILD.bazel
@@ -4,7 +4,10 @@
 
 proto_library(
     name = "api_proto",
-    srcs = ["agent.proto"],
+    srcs = [
+        "agent.proto",
+        "takeover.proto",
+    ],
     visibility = ["//visibility:public"],
 )