Fix missing argument warning