diff --git a/research/attention_ocr/python/model_test.py b/research/attention_ocr/python/model_test.py
index c3fd52180acb9339c8d74af4f6f9cd53ccab99fa..b9576f7a13640aab6f8401319752164a1addabf5 100644
--- a/research/attention_ocr/python/model_test.py
+++ b/research/attention_ocr/python/model_test.py
@@ -16,6 +16,7 @@
 """Tests for the model."""
 
 import numpy as np
+from six.moves import xrange
 import string
 import tensorflow as tf
 from tensorflow.contrib import slim