Small change to be able to use python 3