add and fix test