Error message bug fix