Fix argument mismatch