Sorted map in RemoveMinatnenace response

1 files changed, 2 insertions, 3 deletions

diff --git a/yt/yt/client/api/rpc_proxy/client_impl.cpp b/yt/yt/client/api/rpc_proxy/client_impl.cpp
index 33451d29049..eb77184dc02 100644
--- a/yt/yt/client/api/rpc_proxy/client_impl.cpp
+++ b/yt/yt/client/api/rpc_proxy/client_impl.cpp
@@ -1862,9 +1862,8 @@ TFuture<TMaintenanceCounts> TClient::RemoveMaintenance(
             counts[EMaintenanceType::DisableTabletCells] = rspValue->disable_tablet_cells();
             counts[EMaintenanceType::PendingRestart] = rspValue->pending_restart();
         } else {
-            for (auto type : TEnumTraits<EMaintenanceType>::GetDomainValues()) {
-                auto it = protoCounts.find(ConvertMaintenanceTypeToProto(type));
-                counts[type] = it == protoCounts.end() ? 0 : it->second;
+            for (auto [type, count] : protoCounts) {
+                counts[CheckedEnumCast<EMaintenanceType>(type)] = count;
             }
         }