Error check bug fix