#4 fix test