Fix for unsaved edit warning