Show search section from "Product Search" extension in Storefront header

functions.php

// Hide native search bar
remove_action( 'storefront_header', 'storefront_product_search', 	40 );

// Show "Product Search" search bar
add_action( 'storefront_header', 'storefront_predictive_search', 	40 );

function storefront_predictive_search() {
  echo do_shortcode('[woocommerce_product_search]');
}

style.css

.product-search.floating {
    width: 21.7391304348%;
    float: right;
    margin-right: 0;
    clear: none;
}