diff --git a/src/lib.rs b/src/lib.rs
index c54a90b..9330627 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,4 +1,4 @@
-#![feature(btree_drain_filter)]
+#![feature(btree_extract_if)]
 
 pub mod args;
 pub mod counter;
diff --git a/src/main.rs b/src/main.rs
index 592f022..9e7cf8a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,4 @@
-#![feature(btree_drain_filter)]
+#![feature(btree_extract_if)]
 
 mod args;
 mod counter;
diff --git a/src/seg_tree.rs b/src/seg_tree.rs
index 62f48d1..7f87e81 100644
--- a/src/seg_tree.rs
+++ b/src/seg_tree.rs
@@ -113,7 +113,7 @@ fn infer_namespaces_aux(h: &mut BTreeSet<NamespaceCandidate>, MIN_NS_SIZE: usize
         let mut found = false;
 
         match h
-            .drain_filter(|item| {
+            .extract_if(|item| {
                 if !found {
                     let suitable = item
                         .node