diff --git a/wpimath/build.gradle b/wpimath/build.gradle
index 09df80c99a..3496ba7f98 100644
--- a/wpimath/build.gradle
+++ b/wpimath/build.gradle
@@ -20,8 +20,14 @@ cppHeadersZip {
     from('src/main/native/thirdparty/gcem/include') {
         into '/'
     }
+    from("$buildDir/generated/source/proto/main/cpp") {
+        into '/wpimath/protobuf'
+        include '*.h'
+    }
 }
 
+cppHeadersZip.dependsOn generateProto
+
 model {
     components {
         all {