Fix function signature