diff --git a/tensor2tensor/models/multimodel.py b/tensor2tensor/models/multimodel.py
index 60f098e5e..b42d71cb3 100644
--- a/tensor2tensor/models/multimodel.py
+++ b/tensor2tensor/models/multimodel.py
@@ -27,6 +27,7 @@
 from tensor2tensor.utils import registry
 from tensor2tensor.utils import t2t_model
 
+from six.moves import xrange  # pylint: disable=redefined-builtin
 import tensorflow as tf