Fixed a small CLI bug