bug fix and error message update