Fixed broken tests due to changed message format.