diff --git a/addressfield.tokens.inc b/addressfield.tokens.inc
index 15c5774..cc5ff63 100644
--- a/addressfield.tokens.inc
+++ b/addressfield.tokens.inc
@@ -106,8 +106,8 @@ function addressfield_token_info_alter(&$data) {
  * Implements hook_tokens().
  */
 function addressfield_tokens($type, $tokens, array $data = array(), array $options = array()) {
-  if (isset($options['language'])) {
-    $language_code = $options['language']->language;
+  if (isset($data['entity']->language)) {
+    $language_code = $data['entity']->language;
   }
   else {
     $language_code = LANGUAGE_NONE;