updated to work with WooCommerce 2.0.2