diff --git a/onert-micro/onert-micro/src/train/losses_functions/SparseCrossEntropy.cpp b/onert-micro/onert-micro/src/train/losses_functions/SparseCrossEntropy.cpp
index 4a194c5edbd..55870b86055 100644
--- a/onert-micro/onert-micro/src/train/losses_functions/SparseCrossEntropy.cpp
+++ b/onert-micro/onert-micro/src/train/losses_functions/SparseCrossEntropy.cpp
@@ -38,6 +38,13 @@ void SparseCrossEntropy::calculateErrorBackpropagation(const uint32_t flat_size,
 
   for (uint32_t i = 0; i < flat_size; ++i)
   {
-    output_grad[i] = (calculated_data[i] + float(10.0e-32)) - (i == label);
+    if (i == label)
+    {
+      output_grad[i] = -1.0 / (calculated_data[i] + float(10.0e-20));
+    }
+    else
+    {
+      output_grad[i] = 0.0;
+    }
   }
 }