diff --git a/nomad/structs/generate.sh b/nomad/structs/generate.sh
index 58739e7ee..0d799fed3 100755
--- a/nomad/structs/generate.sh
+++ b/nomad/structs/generate.sh
@@ -2,5 +2,9 @@
 set -e
 
 FILES="$(ls ./*.go | grep -v -e _test.go -e .generated.go | tr '\n' ' ')"
-codecgen -d 100 -t codegen_generated -o structs.generated.go ${FILES}
-sed -i'' -e 's|"github.com/ugorji/go/codec|"github.com/hashicorp/go-msgpack/codec|g' structs.generated.go
+codecgen \
+    -c github.com/hashicorp/go-msgpack/codec \
+    -d 100 \
+    -t codegen_generated \
+    -o structs.generated.go \
+    ${FILES}