A quick hack to work with python 3