Fix missing import.

Change-Id: I04b826b4e54c27eeff567358c0e50315c90c515e