diff --git a/wal/genproto.sh b/wal/genproto.sh
new file mode 100755
index 000000000..1aabc60c2
--- /dev/null
+++ b/wal/genproto.sh
@@ -0,0 +1,8 @@
+set -e
+protoc --gogo_out=. -I=.:$GOPATH/src/code.google.com/p/gogoprotobuf/protobuf:$GOPATH/src *.proto
+
+prefix=github.com/coreos/etcd/third_party
+sed \
+	-i'.bak' \
+	"s|code.google.com/p/gogoprotobuf/proto|$prefix/code.google.com/p/gogoprotobuf/proto|" *.go
+rm *.bak
diff --git a/wal/record.pb.go b/wal/record.pb.go
index c70e04ee5..b4fab1f21 100644
--- a/wal/record.pb.go
+++ b/wal/record.pb.go
@@ -13,14 +13,14 @@
 */
 package wal
 
-import proto "code.google.com/p/gogoprotobuf/proto"
+import proto "github.com/coreos/etcd/third_party/code.google.com/p/gogoprotobuf/proto"
 import json "encoding/json"
 import math "math"
 
 // discarding unused import gogoproto "code.google.com/p/gogoprotobuf/gogoproto/gogo.pb"
 
 import io "io"
-import code_google_com_p_gogoprotobuf_proto "code.google.com/p/gogoprotobuf/proto"
+import code_google_com_p_gogoprotobuf_proto "github.com/coreos/etcd/third_party/code.google.com/p/gogoprotobuf/proto"
 
 // Reference proto, json, and math imports to suppress error if they are not otherwise used.
 var _ = proto.Marshal