Fixed bug in cli.py