Fix sample test error for CI testing