import the required class

Change-Id: Ia315c81efd865ad7c9766eaf0aacf5b794190773