fix for python 2