From 4a0e8aa78078f85b7dbac9605818bbe314bf70b3 Mon Sep 17 00:00:00 2001
From: zandercymatics <141044360+zandercymatics@users.noreply.github.com>
Date: Wed, 15 Nov 2023 15:06:56 -0700
Subject: [PATCH] Fix attr err

---
 src/epplibwrapper/socket.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/epplibwrapper/socket.py b/src/epplibwrapper/socket.py
index 944dd2dfc..4199e91e5 100644
--- a/src/epplibwrapper/socket.py
+++ b/src/epplibwrapper/socket.py
@@ -82,7 +82,10 @@ class Socket:
                 response = self.client.send(self.login)
             except (LoginError, OSError) as err:
                 logger.error(err)
-                if err.should_retry() and counter < 10:
+                should_retry = True
+                if isinstance(err, LoginError):
+                    should_retry = err.should_retry()
+                if should_retry and counter < 10:
                     counter += 1
                     sleep((counter * 50) / 1000)  # sleep 50 ms to 150 ms
                 else:  # don't try again