diff --git a/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java b/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java
index 2244e250..1fa2334a 100644
--- a/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java
+++ b/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java
@@ -790,6 +790,12 @@ private boolean shouldRetryException(Exception e) {
             return true;
         }
 
+        if (e instanceof IOException) {
+            return (e.getMessage().contains("unexpected end of stream") ||
+                    e.getMessage().contains("timeout") ||
+                    e.getMessage().contains("connection refused"));
+        }
+
         if (e instanceof RuntimeException) {
             String message = e.getMessage();
             return message != null && (