Fix the build broken by the previous commit