Fix function declaration