Fix type, follow-up r81615