Add missing class from previous commit.