Fix visibility issue with the `edit` button