diff --git a/src/torchcachex/backend.py b/src/torchcachex/backend.py
index b5b78c6..7012145 100644
--- a/src/torchcachex/backend.py
+++ b/src/torchcachex/backend.py
@@ -592,8 +592,7 @@ def _flush_segment(self) -> None:
 
         # Create RecordBatch from arrays (not pydict, to handle pre-converted arrays)
         record_batch = pa.RecordBatch.from_arrays(
-            list(arrays.values()),
-            schema=self.schema
+            list(arrays.values()), schema=self.schema
         )
 
         # 2. Write to temp Arrow file
diff --git a/tests/test_recovery.py b/tests/test_recovery.py
index b6c38c3..ea619dc 100644
--- a/tests/test_recovery.py
+++ b/tests/test_recovery.py
@@ -191,7 +191,7 @@ def test_atomic_flush(self):
             # Count entries in segments
             segment_count = 0
             for segment_file in segments_dir.glob("segment_*.arrow"):
-                with pa.memory_map(str(segment_file), 'r') as source:
+                with pa.memory_map(str(segment_file), "r") as source:
                     reader = pa.ipc.open_file(source)
                     table = reader.read_all()
                     segment_count += len(table)