diff --git a/message_subscribe_ui/message_subscribe_ui.module b/message_subscribe_ui/message_subscribe_ui.module
index 4e8e06d..cc0181f 100644
--- a/message_subscribe_ui/message_subscribe_ui.module
+++ b/message_subscribe_ui/message_subscribe_ui.module
@@ -159,7 +159,7 @@ function message_subscribe_ui_tab($account, $flag_name = NULL) {
   // Change the flag's relationship to point to our flag.
   $relationships = $view->display_handler->get_option('relationships');
   // Determine if a relationship was set.
-  $rel_set = FALSE;
+
   foreach ($relationships as $key => $relationship) {
     if (strpos($key, 'flag_') === 0) {
       // Check that the flag is valid.
@@ -169,12 +169,9 @@ function message_subscribe_ui_tab($account, $flag_name = NULL) {
         watchdog('message_sbuscribe', t('Flag %name is not setup corrcetly. It is probably disabled or have no bundles configured.'), $params);
         return drupal_set_message(t('There Flag %name is not configured correctly.', $params), 'error');
       }
-
-      if (!$rel_set) {
-        $rel_set = TRUE;
-        $relationships[$key]['flag'] = $flag_name;
-        $view->display_handler->set_option('relationships', $relationships);
-      }
+      $rel_set = TRUE;
+      $relationships[$key]['flag'] = $flag_name;
+      $view->display_handler->set_option('relationships', $relationships);
     }
   }