From 447107834c5f9411209a80d35a0a5cf60f079c18 Mon Sep 17 00:00:00 2001
From: xds <iam@voroninv.ru>
Date: Tue, 3 Feb 2026 09:46:30 +0300
Subject: [PATCH] + fixes

---
 routers/gen_router.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/routers/gen_router.py b/routers/gen_router.py
index 24c8041..e2dd856 100644
--- a/routers/gen_router.py
+++ b/routers/gen_router.py
@@ -143,7 +143,7 @@ async def change_quality(call: CallbackQuery, state: FSMContext, dao: DAO):
 async def gen_mode_change_type(call: CallbackQuery, state: FSMContext, dao: DAO):
     await call.answer()
     data = await state.get_data()
-    if GenType(data['type']) is GenType.IMAGE:
+    if GenType[data['type']] is GenType.IMAGE:
         await state.update_data({"type": GenType.TEXT.name})
     else:
         await state.update_data({"type": GenType.IMAGE.name})