Fixing bug on edition