fix test
small tweak