just fix test