Fix test to expect new UA header