Fixed editing bug