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})