Fix up sample code