Fix test expectations