Fix sample code