1 files changed, 2 insertions, 9 deletions

diff --git a/cogs/cron.py b/cogs/cron.py
index b4f494a..aad43d2 100644
--- a/cogs/cron.py
+++ b/cogs/cron.py
@@ -30,13 +30,10 @@ class Cron(commands.Cog):
 
     def load_tasks(self):
         try:
-            conn = get_connection()
-            try:
+            with get_connection() as conn:
                 c = conn.cursor()
                 c.execute("SELECT id, cron_expression, type, channel_id, active, audio_filename, message FROM cron_jobs")
                 rows = c.fetchall()
-            finally:
-                conn.close()
 
             self.scheduled_tasks = [
                 {
@@ -57,8 +54,7 @@ class Cron(commands.Cog):
 
     def save_tasks(self):
         try:
-            conn = get_connection()
-            try:
+            with get_connection() as conn:
                 c = conn.cursor()
                 c.execute("PRAGMA foreign_keys = ON")
 
@@ -78,10 +74,7 @@ class Cron(commands.Cog):
                             task.get("message"),
                         ),
                     )
-
                 conn.commit()
-            finally:
-                conn.close()
         except Exception as e:
             logger.error(f"Error saving cron jobs: {e}")