Fix fuckup from last commit