Fix wrong class name

Change-Id: I3d81505748eb03f4e723dbb40265740a3424323c