add missing returns