Update test_cli.py