Fix missing argument preventing compilation