Correcting  edit bug