Add missing return keyword in example code.

Missed a spot :)