#119 fix test