Fix lacking include in previous pull request.