Minor change for python 3 compatability