diff --git a/src/galois/_fields/_array.py b/src/galois/_fields/_array.py index 9f1173f32..2895a4563 100644 --- a/src/galois/_fields/_array.py +++ b/src/galois/_fields/_array.py @@ -989,7 +989,7 @@ def additive_order(self) -> int | np.ndarray: if x.ndim == 0: order = 1 if x == 0 else field.characteristic else: - order = field.characteristic * np.ones(x.shape, dtype=np.int64) + order = field.characteristic * np.ones(x.shape, dtype=field.dtypes[-1]) order[np.where(x == 0)] = 1 return order