fix test not passing