Bug with edit button on cell fixed