Fix test expectations

TBR=ojan@chromium.org

Review URL: https://codereview.chromium.org/800103002