Fix function prototype