Fix compilation error in last commit