Fix restoration bug
Work on Iron & Brimstone modes