Fix Edit button