Fix compile after last commit