diff --git a/pyvortex/python/vortex/dataset.py b/pyvortex/python/vortex/dataset.py
index b8455f2b13..f28ed2f27a 100644
--- a/pyvortex/python/vortex/dataset.py
+++ b/pyvortex/python/vortex/dataset.py
@@ -132,7 +132,7 @@ def take(
         use_threads: bool | None = None,
         memory_pool: pa.MemoryPool = None,
     ) -> pa.Table:
-        return self._dataset.to_array(columns, batch_size, filter).take(encoding.array(indices)).to_arrow_table()
+        return self._dataset.to_array(columns, filter).take(encoding.array(indices)).to_arrow_table()
 
     def to_record_batch_reader(
         self,
@@ -211,7 +211,7 @@ def to_table(
         del memory_pool
         if filter is not None:
             filter = arrow_to_vortex_expr(filter, self.schema)
-        return self._dataset.to_array(columns, batch_size, filter).to_arrow_table()
+        return self._dataset.to_array(columns, filter).to_arrow_table()
 
 
 class VortexScanner(pa.dataset.Scanner):