FIxed a bug with messages showing incorrect category