diff -u b/tracker2.module b/tracker2.module
--- b/tracker2.module
+++ b/tracker2.module
@@ -209,9 +209,8 @@
   }
 
   // Flag integration.
-  if ($flag_name = tracker2_get_follow_flag()) {
-    $flag = flag_get_flag($flag_name);
-    $flag->flag('flag', $nid, user_load($uid));
+  if ($flag = tracker2_get_follow_flag()) {
+    flag('flag', $flag, $nid, user_load($uid));
   }
 }
 
@@ -278,9 +277,8 @@
   }
 
   // Flag integration.
-  if ($flag_name = tracker2_get_follow_flag()) {
-    $flag = flag_get_flag($flag_name);
-    $flag->flag('unflag', $nid, user_load($uid));
+  if ($flag = tracker2_get_follow_flag()) {
+    flag('unflag', $flag, $nid, user_load($uid));
   }
 }