diff --git a/dcor/_dcor_internals.py b/dcor/_dcor_internals.py
index 21b390d..4ddbf7a 100644
--- a/dcor/_dcor_internals.py
+++ b/dcor/_dcor_internals.py
@@ -343,7 +343,7 @@ def u_centered(a: Array, *, out: Array | None = None) -> Array:
 
     # The diagonal is zero
     xp = array_namespace(a)
-    out[xp.eye(dim, dtype=xp.bool)] = 0
+    out[xp.eye(dim, dtype=bool)] = 0
 
     return out
 
diff --git a/dcor/_energy.py b/dcor/_energy.py
index dc4eee7..c935a1f 100644
--- a/dcor/_energy.py
+++ b/dcor/_energy.py
@@ -78,7 +78,7 @@ def _check_valid_energy_exponent(exponent: float) -> None:
 def _get_flat_upper_matrix(x: Array, k: int) -> Array:
     """Get flat upper matrix from diagonal k."""
     xp = array_namespace(x)
-    x_mask = xp.triu(xp.ones_like(x, dtype=xp.bool), k=k)
+    x_mask = xp.triu(xp.ones_like(x, dtype=bool), k=k)
     x_mask_flat = xp.reshape(x_mask, -1)
     x_flat = xp.reshape(x, -1)