Fix failing test with the message helper