Fix minor test issue