diff --git a/linear_algebra/src/power_iteration.py b/linear_algebra/src/power_iteration.py
index 83c2ce48c3a0..2bbe02de2d0f 100644
--- a/linear_algebra/src/power_iteration.py
+++ b/linear_algebra/src/power_iteration.py
@@ -67,7 +67,7 @@ def power_iteration(
         lambda_ = np.dot(vector_h, np.dot(input_matrix, vector))
 
         # Check convergence.
-        error = np.abs(lambda_ - lambda_previous) / lambda_
+        error = np.abs(lambda_ - lambda_previous) / np.abs(lambda_)
         iterations += 1
 
         if error <= error_tol or iterations >= max_iterations: