Fixed an issue with the edit mode