Update to be compatible with latest WooCommerce