Small fix for python 3