fix missing return in code example