fix the edit part