diff --git a/skyflow/vault/_client.py b/skyflow/vault/_client.py
index 785c42e..1c1236c 100644
--- a/skyflow/vault/_client.py
+++ b/skyflow/vault/_client.py
@@ -71,7 +71,7 @@ def insert(self, records: dict, options: InsertOptions = InsertOptions()):
                     raise SkyflowError(SkyflowErrorCodes.SERVER_ERROR, SkyflowErrorMessages.BATCH_INSERT_FAILURE.value, result, interface=interface)
                 log_info(InfoMessages.INSERT_DATA_SUCCESS.value, interface)
                 return result
-            except requests.exceptions.ConnectionError as err:
+            except Exception as err:
                 if attempt < max_retries:
                     continue
                 else: