diff --git a/src/main/java/com/databricks/jdbc/exception/DatabricksSQLException.java b/src/main/java/com/databricks/jdbc/exception/DatabricksSQLException.java
index 7353cb254..ed84e8787 100644
--- a/src/main/java/com/databricks/jdbc/exception/DatabricksSQLException.java
+++ b/src/main/java/com/databricks/jdbc/exception/DatabricksSQLException.java
@@ -27,14 +27,14 @@ public DatabricksSQLException(String reason, Throwable cause, String sqlState) {
     this(reason, sqlState, DatabricksVendorCode.getVendorCode(cause), cause);
   }
 
-  // This constructor is used to export chunk download failure logs
-  // TODO : Check chunk retry telemetry logic
+  // Chunk-download path: routes statementId + chunkIndex into the telemetry record so failures
+  // can be bucketed per chunk. Other constructors do not carry chunkIndex.
   public DatabricksSQLException(
       String reason, Throwable cause, String statementId, Long chunkIndex, String sqlState) {
     super(reason, sqlState, DatabricksVendorCode.getVendorCode(cause), cause);
     exportFailureLog(
         DatabricksThreadContextHolder.getConnectionContext(),
-        DatabricksDriverErrorCode.CONNECTION_ERROR.name(),
+        sqlState,
         reason,
         statementId,
         chunkIndex,