diff --git a/src/main.rs b/src/main.rs
index 9027780..51e757f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -246,7 +246,11 @@ async fn button_edit_callback(
 
     let lang = "ru".to_string();
     dialogue
-        .update(State::EditTextOnly { literal, lang })
+        .update(State::Edit {
+            literal,
+            lang,
+            is_caption_set: false,
+        })
         .await?;
 
     bot.send_message(q.from.id, "Send text of button").await?;