fix test a bit