improved the update method in admin class

claudiosanches · Jan 15, 2014 · 31b7d9a · 31b7d9a
1 parent 6a4fcc0
commit 31b7d9a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/admin/class-wc-payment-discounts-admin.php b/admin/class-wc-payment-discounts-admin.php
@@ -71,7 +71,7 @@ protected function init() {
 			}
 
 			// Plugin need update.
-			if ( ! get_option( 'woocommerce_payment_discounts_version' ) ) {
+			if ( ! version_compare( get_option( 'woocommerce_payment_discounts_version' ), WC_Payment_Discounts::VERSION, '>=' ) ) {
 				add_action( 'admin_notices', array( $this, 'needs_update' ) );
 				$this->update();
 			}
@@ -89,8 +89,8 @@ protected function init() {
 	 */
 	protected function update() {
 		if ( isset( $_GET['wc-payment-discounts-update'] ) && '1' == $_GET['wc-payment-discounts-update'] ) {
-			add_option( 'woocommerce_payment_discounts', array() );
-			add_option( 'woocommerce_payment_discounts_version', WC_Payment_Discounts::VERSION );
+			update_option( 'woocommerce_payment_discounts', array() );
+			update_option( 'woocommerce_payment_discounts_version', WC_Payment_Discounts::VERSION );
 			delete_option( 'wcpaydisc_settings' );
 			delete_option( 'wcpaydisc_gateways' );
 		}