Fix sample for breaking change in master