diff --git a/tests/tools/onert_train/src/onert_train.cc b/tests/tools/onert_train/src/onert_train.cc
index 956f04bdc4e..678d13fc963 100644
--- a/tests/tools/onert_train/src/onert_train.cc
+++ b/tests/tools/onert_train/src/onert_train.cc
@@ -202,13 +202,13 @@ int main(const int argc, char **argv)
     Measure measure;
     std::vector<float> losses(num_expecteds);
     phases.run("EXECUTE", [&](const benchmark::Phase &, uint32_t) {
-      const int num_sample = data_length / tri.batch_size;
+      const int num_step = data_length / tri.batch_size;
       const int num_epoch = args.getEpoch();
-      measure.set(num_epoch, num_sample);
+      measure.set(num_epoch, num_step);
       for (uint32_t epoch = 0; epoch < num_epoch; ++epoch)
       {
         std::fill(losses.begin(), losses.end(), 0);
-        for (uint32_t n = 0; n < num_sample; ++n)
+        for (uint32_t n = 0; n < num_step; ++n)
         {
           // get batchsize data
           if (!generator(n, input_data, expected_data))
@@ -248,7 +248,7 @@ int main(const int argc, char **argv)
         std::cout.precision(4);
         for (uint32_t i = 0; i < num_expecteds; ++i)
         {
-          std::cout << "[" << i << "] " << losses[i] / num_sample;
+          std::cout << "[" << i << "] " << losses[i] / num_step;
         }
         std::cout /* << "- accuracy: " << accuracy*/ << std::endl;
       }