Added missing line from last commit