diff --git a/routers/gen_router.py b/routers/gen_router.py
index 9baa519..116b3a2 100644
--- a/routers/gen_router.py
+++ b/routers/gen_router.py
@@ -273,7 +273,7 @@ async def gen_mode_start(
 
     data = await state.get_data()
     try:
-        if GenType(data['type']) is GenType.IMAGE:
+        if GenType[data['type']] is GenType.IMAGE:
             generated_files = await generate_image(
                 prompt=prompt,
                 media=media_ids,  # Передаем список (пустой или с 1 фото)