diff --git a/EntityDependencyIterator.inc b/EntityDependencyIterator.inc
diff --git a/README.txt b/README.txt
diff --git a/entity_dependency.api.php b/entity_dependency.api.php
diff --git a/entity_dependency.core.inc b/entity_dependency.core.inc
diff --git a/entity_dependency.module b/entity_dependency.module
index 982d9ff..15fa428 100644
--- a/entity_dependency.module
+++ b/entity_dependency.module
@@ -8,12 +8,24 @@ module_load_include('inc', 'entity_dependency', 'entity_dependency.core');
 /**
  * Factory function for an entity dependency iterator.
  *
- * The $entities array should be structured as below, where all string keys
- * are entity types and the numeric keys are entity ids.
+ * The $entities array should be structured as below, as an array of arrays
+ * with keys:
+ * - type: The entity type.
+ * - id: The entity id.
  * @code
  *   $entities = array(
- *     'node' => array(10, 12),
- *     'taxonomy_term' => array(16),
+ *     array(
+ *       'type' => 'node',
+ *       'id' => 10,
+ *     ),
+ *     array(
+ *       'type' => 'node',
+ *       'id' => 12,
+ *     ),
+ *     array(
+ *       'type' => 'taxonomy_term',
+ *       'id' => 16,
+ *     ),
  *   );
  * @endcode
  */