add small test fix