Fix restoration example imports