Fix bug in cli.py