Fix sample test