fixing issue with arguments in the cli version