Add missing includes.

Test: mma
Bug: None
Change-Id: I8a3390d186aec73f55ae2be2c7b4b2deffed810f