handle trait items as well

1 files changed, 1 insertions, 0 deletions

diff --git a/src/librustc_mir/transform/add_validation.rs b/src/librustc_mir/transform/add_validation.rs
index 578a63e44b0..374b658dfc7 100644
--- a/src/librustc_mir/transform/add_validation.rs
+++ b/src/librustc_mir/transform/add_validation.rs
@@ -112,6 +112,7 @@ fn fn_contains_unsafe<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>, src: MirSource) ->
     match tcx.hir.find(fn_node_id) {
         Some(Node::NodeItem(item)) => finder.visit_item(item),
         Some(Node::NodeImplItem(item)) => finder.visit_impl_item(item),
+        Some(Node::NodeTraitItem(item)) => finder.visit_trait_item(item),
         Some(Node::NodeExpr(item)) => {
             // This is a closure.
             // We also have to walk up the parents and check that there is no unsafe block