diff --git a/flag.module b/flag.module
index 5c26b7a..76616f7 100644
--- a/flag.module
+++ b/flag.module
@@ -1825,6 +1825,14 @@ function flag_get_flags($entity_type = NULL, $content_subtype = NULL, $account =
     }
   }
 
+  if (module_exists('og_flag')) {
+    foreach ($flags as $flag) {
+      if (method_exists($flag, 'set_context')) {
+        $flag->set_context();
+      }
+    }
+  }
+
   // Make a variable copy to filter types and account.
   $filtered_flags = $flags;