[FIX] Pass partner_id as a kwarg