Fixed the edit button