Fixed not working for edits.