Fixing an import problem

Change-Id: I17114cdb58505f8d62419edbfb6f7eb1d8d664cc