Fix sample bug