Include headers to get prototype

Change-Id: Ia68c9dbffe7f745f8503ed476e8734fdd4404122