small code change for python 3