Woocommerce Добавить в корзину перенаправить на Checkout не работает для продукта с вариантами

Я пытаюсь перенаправить людей, которые нажимают кнопку «Добавить в корзину», прямо на страницу оформления заказа. Только кнопка «Добавить в корзину» добавляет товар в корзину, но не перенаправляет его вообще.

Код, который я использую, работает только для продукта без изменений.

Я использую последнюю версию woocommerce 3.0.1.

//REDIRECT TO CHECKOUT

add_filter ('add_to_cart_redirect', 'redirect_to_checkout');

function redirect_to_checkout() {
return WC()->cart->get_checkout_url();
}