Fixed reset after editing bug