From a0f77cbba537514f84f64c5be660f2bdcd69782f Mon Sep 17 00:00:00 2001
From: SoniEx2 <endermoneymod@gmail.com>
Date: Thu, 22 Apr 2021 11:10:09 -0300
Subject: Fix things not being cleaned up correctly

---
 src/lib.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/lib.rs')

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) => {
-- 
cgit 1.4.1