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())