Fix message assertion in integration test