Fix edit button hiding