Adding missing class from previous commit