diff --git a/dvc/tree/pool.py b/dvc/tree/pool.py
index 0b223373e..9bff33ca4 100644
--- a/dvc/tree/pool.py
+++ b/dvc/tree/pool.py
@@ -11,7 +11,7 @@ def get_connection(conn_func, *args, **kwargs):
     conn = pool.get_connection()
     try:
         yield conn
-    except BaseException:
+    except Exception:
         conn.close()
         raise
     else: