Missing #include

Change-Id: Ib78efbb40f95884e8446bf247265004fe66a966b