diff --git a/protos/db_adapter.proto b/protos/db_adapter.proto
index 3760c672a..f254ee321 100644
--- a/protos/db_adapter.proto
+++ b/protos/db_adapter.proto
@@ -2,6 +2,8 @@ syntax = "proto3";
 
 package dataform;
 
+option go_package = "github.com/dataform-co/dataform/protos/dataform";
+
 import "core.proto";
 
 // Common structures for Evaluation, execution and compilation.
diff --git a/protos/jit.proto b/protos/jit.proto
index ca0d0707e..4807c2f28 100644
--- a/protos/jit.proto
+++ b/protos/jit.proto
@@ -2,6 +2,8 @@ syntax = "proto3";
 
 package dataform;
 
+option go_package = "github.com/dataform-co/dataform/protos/dataform";
+
 import "google/protobuf/empty.proto";
 import "google/protobuf/struct.proto";
 import "google/protobuf/timestamp.proto";