Fix in cli.py --restore utility