fix bug and update error message