Thanks for the patch! You also need to add `from __future__ import print_function`.