add back the CLI unit test