Restoration now working as intended