Small fix from last commit