Fix an import

Change-Id: I6234127e0d6e27729310b6985bead4a232b4373d