Add missing import

Change-Id: Id5012edaa2da7528e9a47db521ee92297678ec58