edit mode fix