Fix edit bugs