diff --git a/unittest/python/test_tensor.py b/unittest/python/test_tensor.py
index 8c6775f477bc4dcf40cad250963bf046c3f4e5ce..b82d34f91fb57877e628cce4842ebbbd31312955 100644
--- a/unittest/python/test_tensor.py
+++ b/unittest/python/test_tensor.py
@@ -1,3 +1,5 @@
+from __future__ import print_function
+
 import numpy as np
 import tensor