Fix regression from last commit...