Fix restore_mode bug