#21 fix test