Fix not imported class

Change-Id: I86aa440bbb6260eea5d6f32ac54860e2983a95eb