diff --git a/pilota-build/src/lib.rs b/pilota-build/src/lib.rs
index 963e213..76bf2c0 100644
--- a/pilota-build/src/lib.rs
+++ b/pilota-build/src/lib.rs
@@ -165,8 +165,8 @@ impl<MkB, P> Builder<MkB, P> {
         self
     }
 
-    pub fn with_split(mut self) -> Self {
-        self.split = true;
+    pub fn split_generated_files(mut self, split: bool) -> Self {
+        self.split = split;
         self
     }
 
diff --git a/pilota-build/src/test/mod.rs b/pilota-build/src/test/mod.rs
index f3aae9b..ff8bf1d 100644
--- a/pilota-build/src/test/mod.rs
+++ b/pilota-build/src/test/mod.rs
@@ -133,7 +133,7 @@ fn test_thrift_with_split(
     test_with_split_builder(source, target, gen_dir, |source, target| {
         crate::Builder::thrift()
             .ignore_unused(false)
-            .with_split()
+            .split_generated_files(true)
             .compile_with_config(
                 vec![IdlService::from_path(source.to_owned())],
                 crate::Output::File(target.into()),