Fixed issue with sample

Fixed issue with sample