Add a missing class

Change-Id: Ibfce8077c2e284101a80b0fcb11bf44fce91ecb1