Add missing returns