Fix restoring