Forget to add the callback for the new message types