diff --git a/arithmetic_analysis/jacobi_iteration_method.py b/arithmetic_analysis/jacobi_iteration_method.py
index 21317245e..5c23b1e8a 100644
--- a/arithmetic_analysis/jacobi_iteration_method.py
+++ b/arithmetic_analysis/jacobi_iteration_method.py
@@ -14,7 +14,7 @@ def jacobi_iteration_method(
     constant_matrix: NDArray[float64],
     init_val: list[int],
     iterations: int,
-) -> list[float]:
+) -> NDArray[float64]:
     """
     Jacobi Iteration Method:
     An iterative algorithm to determine the solutions of strictly diagonally dominant