Added missing import

Change-Id: Ia955f3813c5ab4fd40cfada5b7d7152c2041335f