Fix function definition