diff --git a/generate.py b/generate.py
index 4ad9f52..dcd3c1c 100755
--- a/generate.py
+++ b/generate.py
@@ -1,3 +1,4 @@
+from __future__ import print_function
 import numpy as np
 import argparse
 from PIL import Image
@@ -33,6 +34,6 @@
 result = result.transpose(0, 2, 3, 1)
 result = result.reshape((result.shape[1:]))
 result = np.uint8(result)
-print time.time() - start, 'sec'
+print(time.time() - start, 'sec')
 
 Image.fromarray(result).save(args.out)