Add missing class

Reviewed at https://rbcommons.com/s/twitter/r/1414/