Fix edit btton 2

Signed-off-by: Atanas Kerezov <kerezov.atanas@gmail.com>