Fix bad import

Change-Id: I5a31b017b64dc9206ae61f9c009807b608ec7de9