fix test to pass