Using shs on i18n_taxonomy terms for multilingual websites

I wanted to use this great module on a multilingual website using i18n_taxonomy. I discovered that when using this module as a filter on a view, taxonomy terms were not displayed in the same language of the page.
I did a couple of changes, and now it works. This is what I have made:

1. I have added a 'langcode' setting in the view handler filter:

      global $language_content;
      $element_settings = array(
        'create_new_terms' => FALSE,
        'create_new_levels' => FALSE,
        'required' => FALSE,
        'langcode' => $language_content->language,
      );

2. In shs.module, function shs_term_get_children, I have replaced

$tree = taxonomy_get_tree($vid, $parent, 1);

with the following:

$tree = i18n_taxonomy_get_tree($vid, $settings['langcode'], $parent, 1);

I'm not sure if this is the best way of solving the problem (any feedback will be appreciated), but now it works fine, picking up the terms in the same language of the page.

It would be great if shs could support i18n_taxonomy terms natively.

Thanks!