From 7867a4a8ca864dcc96b7655f02f5fdf37b837102 Mon Sep 17 00:00:00 2001
From: hai007 <hais@google.com>
Date: Thu, 11 Mar 2021 16:29:13 -0800
Subject: [PATCH] Internal change

PiperOrigin-RevId: 362403241
---
 cpp/core/internal/bwu_manager.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cpp/core/internal/bwu_manager.cc b/cpp/core/internal/bwu_manager.cc
index 4a21bd1c..b8b867ae 100644
--- a/cpp/core/internal/bwu_manager.cc
+++ b/cpp/core/internal/bwu_manager.cc
@@ -551,6 +551,7 @@ void BwuManager::RunUpgradeFailedProtocol(
   if (medium_ != Medium::UNKNOWN_MEDIUM) {
     Revert();
   }
+  in_progress_upgrades_.erase(endpoint_id);
 }
 
 bool BwuManager::ReadClientIntroductionFrame(EndpointChannel* channel,