Import a required class

Change-Id: Ifad201ea07bd5c7d4dd4641d886ece8649325dec