Add missing arg