Catch closed connection errors

1 files changed, 1 insertions, 1 deletions

diff --git a/db.py b/db.py
index ed7fa9d..eb7204b 100644
--- a/db.py
+++ b/db.py
@@ -39,7 +39,7 @@ class Db:
     for _ in range(self.MAX_RETRIES):
       try:
         return self.conn.cursor()
-      except psycopg2.OperationalError:
+      except psycopg2.Error:
         self.connect()
 
     raise Exception(f"Failed to connect to database after {Db.MAX_RETRIES} retries.")