Fixed assertion to match current message