diff --git a/src/mito2/src/read.rs b/src/mito2/src/read.rs
index b671152dee..567754f939 100644
--- a/src/mito2/src/read.rs
+++ b/src/mito2/src/read.rs
@@ -1163,7 +1163,7 @@ impl FlatSource {
         }
     }
 
-    pub fn schema(&self) -> &SchemaRef {
+    pub(crate) fn schema(&self) -> &SchemaRef {
         &self.schema
     }
 
@@ -1171,6 +1171,7 @@ impl FlatSource {
         self.inner.next_batch().await
     }
 
+    #[cfg(test)]
     pub(crate) fn take_iter(self) -> BoxedRecordBatchIterator {
         match self.inner {
             FlatSourceInner::Iter(iter) => iter,