Fixed a bugin the CLI