Added missing class from last commit