Fix for updated woocommerce