diff --git a/keras/src/ops/nn_test.py b/keras/src/ops/nn_test.py
index 5e319f93284..6efd71c705d 100644
--- a/keras/src/ops/nn_test.py
+++ b/keras/src/ops/nn_test.py
@@ -1255,7 +1255,9 @@ def test_softmax_correctness_with_axis_tuple(self):
         combination = combinations(range(3), 2)
         for axis in list(combination):
             result = keras.ops.nn.softmax(input, axis=axis)
-            normalized_sum_by_axis = np.sum(np.asarray(result), axis=axis)
+            normalized_sum_by_axis = np.sum(
+                ops.convert_to_numpy(result), axis=axis
+            )
             self.assertAllClose(normalized_sum_by_axis, 1.0)
 
     def test_log_softmax(self):