Fix code sample

Fixed return variable name in the code sample