diff --git a/src/components/MessageInput/EditMessageForm.tsx b/src/components/MessageInput/EditMessageForm.tsx
index 7de9ad88e..e7d5ca246 100644
--- a/src/components/MessageInput/EditMessageForm.tsx
+++ b/src/components/MessageInput/EditMessageForm.tsx
@@ -95,8 +95,12 @@ export const EditMessageForm = <
               )}
             </div>
             <div>
-              <button onClick={clearEditingState}>{t<string>('Cancel')}</button>
-              <button type='submit'>{t<string>('Send')}</button>
+              <button className='str-chat__edit-message-cancel' onClick={clearEditingState}>
+                {t<string>('Cancel')}
+              </button>
+              <button className='str-chat__edit-message-send' type='submit'>
+                {t<string>('Send')}
+              </button>
             </div>
           </div>
         </form>