Data structure to represent possibly unsafe module
diff --git a/macro/src/lib.rs b/macro/src/lib.rs
index bc8d19b..b8a592b 100644
--- a/macro/src/lib.rs
+++ b/macro/src/lib.rs
@@ -13,9 +13,10 @@
 mod syntax;
 mod type_id;
 
+use crate::syntax::file::Module;
 use crate::syntax::namespace::Namespace;
 use proc_macro::TokenStream;
-use syn::{parse_macro_input, ItemMod, LitStr};
+use syn::{parse_macro_input, LitStr};
 
 /// `#[cxx::bridge] mod ffi { ... }`
 ///
@@ -39,7 +40,7 @@
     let _ = syntax::error::ERRORS;
 
     let namespace = parse_macro_input!(args as Namespace);
-    let ffi = parse_macro_input!(input as ItemMod);
+    let ffi = parse_macro_input!(input as Module);
 
     expand::bridge(&namespace, ffi)
         .unwrap_or_else(|err| err.to_compile_error())