Fixup sample code