Fixing edit button