Remove fields from register form but not user profile update

/**
 * This example removes the WP-Members default address/phone fields
 * from the form if it is being displayed for registration, but
 * would leave them in for user profile update or any other 
 * process.
 */
add_filter( 'wpmem_fields', 'my_remove_reg_fields_filter', 10, 2 );
function my_remove_reg_fields_filter( $fields, $tag ) {
 
    if ( 'register' == $tag ) {
        unset( $fields['billing_address_1'] );
        unset( $fields['billing_address_2'] );
        unset( $fields['billing_city']      );
        unset( $fields['billing_state']     );
        unset( $fields['billing_postcode']  );
        unset( $fields['billing_country']   );
        unset( $fields['billing_phone']     );
    }
     
    return $fields;
}