Fixing unittest after changing the message format