fix minor test issue