redirect not logged in users to specific page and redirect back

The problem is not the first redirection (already solved in other case here), the problem is the second redirection back.

No problem with the first redirection, when no logged in user come to page ID 10 (checkout page) he is directly redirect to custom login page but when he log in he is not redirect back to the checkout page, instead we can see before the login process in the url xxx.com/myaccount/?redirect_to=xxx.com/checkout/, he stay in the account page and dont go back automatically to the checkout page.

This is the code I used:

add_action('template_redirect','check_if_logged_in');
function check_if_logged_in()
{
    $pageid = 10; // checkout page
    if(!is_user_logged_in() && is_page($pageid))
    {
        $url = add_query_arg(
            'redirect_to',
            get_permalink($pagid),
            site_url('/myaccount/')
        );
        wp_redirect($url);
        exit;
    }
}

I just need that the second back redirection to the checkout page works.
This is what we need : No logged in user -> checkout page -> login page -> logged in -> checkout page

Any help would be appreciated. Thanks.

https://wordpress.org/plugins/woocommerce/