Fixed return type from sample code

Fixed typo on return type from sample code.