Small fix to CLI module.