Forgot to add in the message class