diff --git a/term_reference_tree.module b/term_reference_tree.module
index 91313ef..1aea8b5 100644
--- a/term_reference_tree.module
+++ b/term_reference_tree.module
@@ -138,7 +138,7 @@ function _term_reference_tree_get_children($tid, $vid) {
   // module is enabled).
   $select = db_select('taxonomy_term_data', 'd');
   $select->join('taxonomy_term_hierarchy', 'h', 'd.tid = h.tid');
-  $result = $select->fields('d', array('tid', 'name'))
+  $result = $select->fields('d', array('tid', 'vid', 'name'))
     ->condition('d.vid', $vid, '=')
     ->condition('h.parent', $tid, '=')
     ->orderBy('weight')