diff --git a/uc_cart/uc_cart.module b/uc_cart/uc_cart.module
index 89cecc6..c840ff3 100644
--- a/uc_cart/uc_cart.module
+++ b/uc_cart/uc_cart.module
@@ -690,6 +690,7 @@ function uc_cart_view_form($form, &$form_state, $items = NULL) {
     '#type' => 'submit',
     '#name' => 'update-cart',
     '#value' => t('Update cart'),
+    '#submit' => array('uc_cart_view_form_submit', 'uc_cart_view_form_update_message'),
   );
   $form['actions']['checkout'] = array(
     '#theme' => 'uc_cart_checkout_buttons',
@@ -725,10 +726,15 @@ function uc_cart_view_form_submit($form, &$form_state) {
       uc_cart_update_item_object((object)$form_state['values']);
     }
   }
+}
 
-  if ($form_state['triggering_element']['#name'] == 'update-cart') {
-    drupal_set_message(t('Your cart has been updated.'));
-  }
+/**
+ * Displays "cart updated" message for uc_cart_view_form().
+ *
+ * @see uc_cart_view_form()
+ */
+function uc_cart_view_form_update_message($form, &$form_state) {
+  drupal_set_message(t('Your cart has been updated.'));
 }
 
 /**