fix test implementation