Fix compilation error from pull request