Add missing callback args