Added support for newer woocommerce