Fix function prototypes

Change-Id: I74596b77f24612aba065a3e862e9cf4e381969df