Small fix for python 3.2