Fix things not being cleaned up correctly

1 files changed, 8 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 9473af0..2a3b90f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -268,6 +268,14 @@ pub fn impl_trait(item: TokenStream) -> TokenStream {
             }
             if found.len() == to_remove.len() {
                 found.push(Vec::new());
+                in_unsafe = false;
+                in_impl = false;
+                in_where = false;
+                in_path = false;
+                in_attr_cont = false;
+                in_generic = false;
+                has_injected_generics = false;
+                count = 0;
             }
             match tt {
                 &TokenTree::Ident(ref ident) => {