Fix workload retries calculation

1 files changed, 5 insertions, 5 deletions

diff --git a/ydb/public/lib/ydb_cli/commands/ydb_workload.cpp b/ydb/public/lib/ydb_cli/commands/ydb_workload.cpp
index 188b1dadc4..c7b2744d65 100644
--- a/ydb/public/lib/ydb_cli/commands/ydb_workload.cpp
+++ b/ydb/public/lib/ydb_cli/commands/ydb_workload.cpp
@@ -136,12 +136,12 @@ void TWorkloadCommand::WorkerFn(int taskId, TWorkloadQueryGenPtr workloadGen, co
                 // }
                 break;
             }
+            if (retryCount > 0) {
+                TotalRetries += retryCount;
+                WindowRetryCount += retryCount;
+            }
+            retryCount = -1;
         }
-        if (retryCount > 0) {
-            TotalRetries += retryCount;
-            WindowRetryCount += retryCount;
-        }
-        retryCount = -1;
         if (it != queryInfoList.end()) {
             continue;
         }