Fix missing arg