Fix update on edit scenarios