From 97d093ff17424f99ee67b56112e91f86d671b3d8 Mon Sep 17 00:00:00 2001
From: Benjamin Saunders <ben.e.saunders@gmail.com>
Date: Sat, 26 Jan 2019 14:22:13 -0800
Subject: [PATCH] Add missing cleanup

This was harmless, but best to be consistent.
---
 quinn-proto/src/endpoint.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/quinn-proto/src/endpoint.rs b/quinn-proto/src/endpoint.rs
index de2c7fde0..997d696a9 100644
--- a/quinn-proto/src/endpoint.rs
+++ b/quinn-proto/src/endpoint.rs
@@ -606,6 +606,7 @@ impl Endpoint {
             .remove(&self.connections[ch].remote());
         self.dirty_timers.remove(&ch);
         self.eventful_conns.remove(&ch);
+        self.needs_transmit.remove(&ch);
         self.connections.remove(ch.0);
     }