Minor fix to unsaved-change catch