diff --git a/bot/handlers/admin/on_off.py b/bot/handlers/admin/on_off.py
index 3378abf..7388c4d 100644
--- a/bot/handlers/admin/on_off.py
+++ b/bot/handlers/admin/on_off.py
@@ -14,4 +14,4 @@ async def on_off_call(message: types.Message, is_command=None):
     db[DBTables.config]['enabled'] = False if message.text.lower() == 'off' else True
     await db[DBTables.config].write()
 
-    await message.reply(f"Generation enabled: {'💚 (yes)' if db[DBTables.config]['enabled'] else '💔 (no)'}")
+    await message.reply(f"Generation enabled: {'💚 (yes)' if db[DBTables.config].get('enabled') else '💔 (no)'}")
diff --git a/bot/handlers/txt2img/txt2img.py b/bot/handlers/txt2img/txt2img.py
index 67de14e..001dcb7 100644
--- a/bot/handlers/txt2img/txt2img.py
+++ b/bot/handlers/txt2img/txt2img.py
@@ -16,7 +16,7 @@ from bot.utils.errorable_command import wrap_exception
 @throttle(cooldown=30, admin_ids=db[DBTables.config].get('admins'))
 async def generate_command(message: types.Message):
     temp_message = await message.reply("⏳ Enqueued...")
-    if not db[DBTables.config]['enabled']:
+    if not db[DBTables.config].get('enabled'):
         await message.reply('💔 Generation is disabled by admins now. Try again later')
         await temp_message.delete()
         return