Fix test_restoration.py rename.