bug fix + better error message