Added missing return in example