Fixing cli test