Fix error message bug