add missing return to make example run