Fix function prototype

This patch fixes the prototypes with warnings enabled.