diff --git a/immudb_wrapper.py b/immudb_wrapper.py
index 280d38b..751e661 100644
--- a/immudb_wrapper.py
+++ b/immudb_wrapper.py
@@ -347,7 +347,7 @@ def verified_set(
         except RpcError:
             return {'error': format_exc()}
 
-    @retry(possible_exc_details=['Connection timed out'])
+    @retry(possible_exc_details=['Connection timed out', 'Socket closed'])
     def notarize(
         self,
         key: str,
@@ -420,7 +420,7 @@ def notarize_git_repo(
             value=payload,
         )
 
-    @retry(possible_exc_details=['Connection timed out'])
+    @retry(possible_exc_details=['Connection timed out', 'Socket closed'])
     def authenticate(
         self,
         key: Union[str, bytes],