Restore state after save