Fixup, missing import

Change-Id: Iae7376ef7477c451328748d67526bea828de5947