diff --git a/kernel/src/vm/vmar/interval_set.rs b/kernel/src/vm/vmar/interval_set.rs
index 026da4adc..57df6c7da 100644
--- a/kernel/src/vm/vmar/interval_set.rs
+++ b/kernel/src/vm/vmar/interval_set.rs
@@ -100,6 +100,7 @@ where
     ///
     /// If no such item exists, returns [`None`]. Otherwise, returns the item
     /// that contains the point.
+    #[cfg(ktest)]
     pub fn take_one(&mut self, point: &K) -> Option<V> {
         let mut cursor = self
             .btree
@@ -122,6 +123,7 @@ where
     ///
     /// This method returns a draining iterator that removes the items from the
     /// interval set.
+    #[cfg(ktest)]
     pub fn take<'a>(&'a mut self, range: &Range<K>) -> IntervalDrain<'a, K, V> {
         let cursor = self
             .btree