diff --git a/src/rbm/matrix.py b/src/rbm/matrix.py
index 39e010f..53f7b8c 100644
--- a/src/rbm/matrix.py
+++ b/src/rbm/matrix.py
@@ -3,12 +3,12 @@ from typing import TypeAlias
 USE_CUDA = 1
 if USE_CUDA:
 	import cupy as np
-	Mat: TypeAlias = np.array
+	Mat: TypeAlias = np.ndarray
 	def convert(src: Mat):
 		return np.asnumpy(src)
 else:
 	import numpy as np
-	Mat: TypeAlias = np.array
+	Mat: TypeAlias = np.ndarray
 	def convert(src: Mat):
 		return src