Fix what tests expect