Added compatibility with WooCommerce