added minor test improvement