Fix the build from the last commit