Add missing import

Change-Id: Iedb8e0441d96248082837bf499455e0ee51644fb