diff --git a/src/mdio/segy/blocked_io.py b/src/mdio/segy/blocked_io.py
index c811abfa..177d0988 100644
--- a/src/mdio/segy/blocked_io.py
+++ b/src/mdio/segy/blocked_io.py
@@ -254,7 +254,7 @@ def to_segy(
     out_byteorder: ByteOrder,
     file_root: str,
     axis: tuple[int] | None = None,
-) -> NDArray:
+) -> Array:
     """Convert MDIO blocks to SEG-Y parts.
 
     Args: