Added the missing import. This reverts commit c51e725644. Change-Id: Icd63619a12e10dee628e568a6200d429a9936cf0
c51e725644