Fix error message in tests