Fixed missing return that broke test discovery.