fixed minor issue with sample