2 files changed, 3 insertions, 0 deletions
diff --git a/p9/src/protocol/messages.rs b/p9/src/protocol/messages.rs
index b5a03c0..f149e3c 100644
--- a/p9/src/protocol/messages.rs
+++ b/p9/src/protocol/messages.rs
@@ -2,6 +2,8 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+use wire_format_derive::P9WireFormat;
+
 use std::io::{self, ErrorKind, Read, Write};
 use std::mem;
 use std::string::String;
diff --git a/p9/src/protocol/wire_format.rs b/p9/src/protocol/wire_format.rs
index bd3d67b..84408e2 100644
--- a/p9/src/protocol/wire_format.rs
+++ b/p9/src/protocol/wire_format.rs
@@ -207,6 +207,7 @@ mod test {
     use std::io::Cursor;
     use std::mem;
     use std::string::String;
+    use wire_format_derive::P9WireFormat;
 
     #[test]
     fn integer_byte_size() {