From ce8478e1505b0f37f6b4e8e432dbf0b89a4b8c6e Mon Sep 17 00:00:00 2001
From: Michal Charemza <michal@charemza.name>
Date: Fri, 17 Jan 2020 21:15:45 +0000
Subject: [PATCH] (feat) Catch more specific exception

---
 dnsrewriteproxy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dnsrewriteproxy.py b/dnsrewriteproxy.py
index d2fb4ca..29d2be5 100644
--- a/dnsrewriteproxy.py
+++ b/dnsrewriteproxy.py
@@ -82,7 +82,7 @@ def DnsProxy(
             for upstream_task in upstream_worker_tasks:
                 try:
                     await upstream_task
-                except Exception:
+                except CancelledError:
                     pass
 
     async def upstream_worker(sock, resolve, upstream_queue):