diff --git a/databuild/BUILD.bazel b/databuild/BUILD.bazel
index cc0dfd1..d95a413 100644
--- a/databuild/BUILD.bazel
+++ b/databuild/BUILD.bazel
@@ -69,7 +69,11 @@ genrule(
 # DataBuild library using generated prost code
 rust_library(
     name = "databuild",
-    srcs = [":generate_databuild_rust"],
+    srcs = [
+        "lib.rs",
+        ":generate_databuild_rust",
+        "structs.rs",
+    ],
     deps = [
         "@crates//:prost",
         "@crates//:prost-types",
diff --git a/databuild/graph/analyze.rs b/databuild/graph/analyze.rs
index 3b8c41b..4f6ae47 100644
--- a/databuild/graph/analyze.rs
+++ b/databuild/graph/analyze.rs
@@ -5,7 +5,7 @@ use std::sync::{Arc, Mutex};
 use std::thread;
 use log::{info, error};
 use simple_logger::SimpleLogger;
-use structs::*;
+use databuild::*;
 
 // Configure a job to produce the desired outputs
 fn configure(job_label: &str, output_refs: &[String]) -> Result<Vec<Task>, String> {
diff --git a/databuild/graph/execute.rs b/databuild/graph/execute.rs
index 502b104..d13213b 100644
--- a/databuild/graph/execute.rs
+++ b/databuild/graph/execute.rs
@@ -1,4 +1,4 @@
-use structs::{JobGraph, Task};
+use databuild::{JobGraph, Task};
 use crossbeam_channel::{Receiver, Sender};
 use log::{debug, error, info, warn};
 use serde::{Deserialize, Serialize};