diff --git a/src/operators/nn/functional/softmax.cairo b/src/operators/nn/functional/softmax.cairo
index 6fdfd747b..bbe9335ec 100644
--- a/src/operators/nn/functional/softmax.cairo
+++ b/src/operators/nn/functional/softmax.cairo
@@ -56,7 +56,7 @@ use debug::PrintTrait;
 /// Cf: NNTrait::softmax docstring
 fn softmaxWide2(z: @Tensor<FP16x16>, axis: usize) -> Tensor<FP16x16> {
     let exp_tensor: Tensor<FP16x16W> = exp_upcast(*z);
-    (*exp_tensor.data.at(0)).print();
+    (exp_tensor.data.len()).print();
 
     // let sum = exp_tensor.reduce_sum(axis, true);
     // (*sum.data.at(0)).print();