diff --git a/parser/src/directives.rs b/parser/src/directives.rs
index 160829e..70c88fc 100644
--- a/parser/src/directives.rs
+++ b/parser/src/directives.rs
@@ -268,10 +268,8 @@
     ) -> Box<dyn Iterator<Item = TokenStream> + 'a> {
         let policy = &config.unsafe_policy;
         match config.unsafe_policy {
-            crate::UnsafePolicy::AllFunctionsSafe => {
-                Box::new(std::iter::once(policy.to_token_stream()))
-            }
             crate::UnsafePolicy::AllFunctionsUnsafe => Box::new(std::iter::empty()),
+            _ => Box::new(std::iter::once(policy.to_token_stream())),
         }
     }
 }