Fix error message
fix test