Fix incorrect import.

Change-Id: Ibd5e78e52953da8fea21e7e7f015a0497cbf9914