Fix error in previous PR.