diff --git a/og_vocab.module b/og_vocab.module
index 679e582..a681b4f 100644
--- a/og_vocab.module
+++ b/og_vocab.module
@@ -1186,13 +1186,18 @@ function og_vocab_form_taxonomy_form_term_alter(&$form, $form_state) {
  *
  * @todo: Take care of administrators, we need to show ALL widgets?
  *
+ * @param $entity_type
+ *  The entity type.
+ * @param $bundle
+ *  The entity bundle.
  * @param $account
  *   User object.
+ *
  * @return
  *   An array with the vocabulary IDs or an empty array if no vocabulary
  *   was found.
  */
-function og_vocab_get_accessible_vocabs($entity_type = NULL, $entity = NULL, $account = NULL) {
+function og_vocab_get_accessible_vocabs($entity_type = NULL, $bundle = NULL, $account = NULL) {
   if (empty($account)) {
     global $user;
     $account = clone $user;
@@ -1214,8 +1219,6 @@ function og_vocab_get_accessible_vocabs($entity_type = NULL, $entity = NULL, $ac
     return;
   }
 
-  list(,, $bundle) = entity_extract_ids($entity_type, $entity);
-
   $query = db_select('og_vocab_relation', 'ogr');
   $query->fields('ogr', array('vid'));