add minor test fix